标签: 模型转换

1 篇文章

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

超多性价比流量卡,扫码查看

这将关闭于 20