大漠yolo环境配置 PT转onnx格式失败

spclonely · 发表于 2024-11-1 10:04:34

新手小白捣鼓一天了，求大神指点

hy520101 · 发表于 2024-11-1 11:53:19

我也一样

hy520101 · 发表于 2024-11-1 11:55:49

系统我都重做了还是这样

spclonely · 发表于 2024-11-1 15:05:48

hy520101 发表于 2024-11-1 11:55
系统我都重做了还是这样

花钱找人解决了安装onnx 1.16.1 版本就行。这方法试了两次就是没试对版本号

hy520101 · 发表于 2024-11-3 13:47:38

spclonely 发表于 2024-11-1 15:05
花钱找人解决了安装onnx 1.16.1 版本就行。这方法试了两次就是没试对版本号 ...

兄弟这个torch版本是多少

spclonely · 发表于 2024-11-4 15:48:09

hy520101 发表于 2024-11-3 13:47
兄弟这个torch版本是多少

我没有报错这个

hy520101 · 发表于 2024-11-4 16:01:45

spclonely 发表于 2024-11-4 15:48
我没有报错这个

我按你的说的修改以后，可以转换模型了，但是识别和训练变成cpu 不用 cua118

spclonely · 发表于 2024-11-5 08:38:33

hy520101 发表于 2024-11-4 16:01
我按你的说的修改以后，可以转换模型了，但是识别和训练变成cpu 不用 cua118 ...

显卡设置那一步有问题吧，
更新下显卡试试
cuda版本不能比N卡版本高

yaoyuanyao · 发表于 2024-11-10 19:18:17

spclonely 发表于 2024-11-1 15:05
花钱找人解决了安装onnx 1.16.1 版本就行。这方法试了两次就是没试对版本号 ...

这个版本在哪里看

spclonely · 发表于 2024-11-11 09:11:59

yaoyuanyao 发表于 2024-11-10 19:18
这个版本在哪里看

直接先卸载然后安装这个版本的试试

		自动登录	找回密码
密码			注册

[易语言] 大漠yolo环境配置 PT转onnx格式失败