Couldn‘t load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompati

乍一看是提示pytorch和torchvision版本的问题。也确实是它们的问题哈 哈 哈。但是有细节要注意了。

昨晚我用我们实验室的3090跑模型,比我的1660台式机还慢!绝了

各种控制变量,最后就是定位到GPU慢,推理慢。

最终,只有换包解决。

首先3090跟1660主机torch版本是一样的,我只能寄希望于3090的pytorch不能很好兼容cuda。毕竟3090出来的时候,cuda差不多已经是11了。

于是去官网下载torch1.8.0+cu111,因为3090本机装了cuda11.1.所以就下了这个,期望包自带的cuda能和我安装的11.1对应起来。之前的版本是torch1.7.0+cu11.0。

这就是需要注意的细节了!

torch 的版本号是1.8.0+cu111

在这里插入图片描述

所以torchvision也得是0.9.0+cu111,而不是0.9.0

我就是直接pip install torchvision==0.9.0,出现题目提到的错误,理所应当以为我的torch就是1.8.0,cu111只是一些附带的东西。

所以结论就是torch和torchvision匹配的版本应该是
torch1.8.0 &&& torchvision 0.9.0
torch1.8.0+cu111 &&& torchvision 0.9.0+cu111

我直接pip install torchvision==0.9.0+cu111找不到包,可能名字有点小区别,直接去pytorch源下的
如果慢了可以去对应的国内镜像源找找

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值