模型训练—-将pth模型转换为onnx 2024-3-28 17:20 | 829 | 模型训练 Github代码 1 安装需要的环境 需要在虚拟环境中安装onnx和onnxruntime(GPU),环境和自己的cuda版本要对应上查询链接 激活环境,查看环境的cuda版本,我是cuda11.6 +cudnn8302,那就选择1.14吧 conda activate xxxx python import torch # 查询cuda版本 pri… 模型转换