RuntimeError: No such operator aten::cudnn_convolution_backward_weight

最近在跑styleGAN_xl的代码,啊花式报错。。。

RuntimeError: No such operator aten::cudnn_convolution_backward_weight

不过我只查到了RuntimeError: No such operator torchvision::nms

(28条消息) RuntimeError: No such operator torchvision::nms的解决方法_牧濑_红莉栖的博客-CSDN博客icon-default.png?t=M5H6https://blog.csdn.net/weixin_43843824/article/details/109448314那么类比一下,是不是我torch和torchvision的版本太高了呢?

赶紧查一查。。。。

pip list

torch                   1.12.0
torchvision             0.13.0

果然,版本太新了。。。我们要降版本,但是这是随便降的么,不是。。。

torch和torchvision对应关系-python黑洞网 (pythonheidong.com)icon-default.png?t=M5H6https://www.pythonheidong.com/blog/article/1153942/aaae1ac8bde1ed599e1c/我python==3.8于是我选择

torch                   1.7.0
torchvision             0.8.1

pip install --no-deps torch==1.7.0
pip install --no-deps torchvision==0.8.1

(28条消息) 解决安装torchvision自动更新torch到最新版本_lxx4610的博客-CSDN博客_升级torch版本icon-default.png?t=M5H6https://blog.csdn.net/lxx4610/article/details/101020815好了,能继续训练了呢,yysd!!!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值