部署框架onnx
1.转成C++程序,输出要考虑清楚
2.cuda的库要匹配,主要还是与自己电脑上的驱动有关系,如果程序卡在cuda加载的时候,优先考虑是否cuda_adv_infer64_8.dll是否加载正确。
3 输入要正确 训练与推理要保持一致
部署onnx
最新推荐文章于 2024-06-04 10:32:43 发布
部署框架onnx
1.转成C++程序,输出要考虑清楚
2.cuda的库要匹配,主要还是与自己电脑上的驱动有关系,如果程序卡在cuda加载的时候,优先考虑是否cuda_adv_infer64_8.dll是否加载正确。
3 输入要正确 训练与推理要保持一致