最近在跑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博客https://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)https://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版本https://blog.csdn.net/lxx4610/article/details/101020815好了,能继续训练了呢,yysd!!!