





如果你暂时无法升级宿主机驱动,可以尝试以下方法绕过 GPU依赖:
export CUDA_VISIBLE_DEVICES=""
# 或者在 Python代码开头加入:
# import os; os.environ["CUDA_VISIBLE_DEVICES"] = ""

根据你贴出的hb_compile.log中的Quantized Cosine可以看出在最后余弦相似度骤降。可以对比一下optimized_float_model.onnx和quantized_model.bc这两个模型在使用相同输入时的下面这三个算子的输出:ArgMax_112,Unsqueeze_113,Cast_114。看他们从哪里开始放大误差,如果是ArgMax_112的类别索引不一致就说明误差发生在输入logits,可以进一步查看Resize_111,Conv_109的输出,如果是在Cast_114不一致,那检查一下数据类型转换数值范围或输出格式。

