问题
问题分析 torchvision的版本有问题,此前安装的torchvision=0.4.1,但是不知道怎么的,在anaconda的pytorch环境里输入
conda list
发现安装的是torchvision==0.4.1
cuda 10.0
pip install torch==1.2.0 torchvision==0.4.0
问题分析及注意事项
我用的cuda 10.0,所以我需要的是torchvision==0.4.0,由于安torchvision的轮子以及我torch的版本也有些奇怪比较复杂,我选择把torch也移除了。重新安装torch和torchvision后问题解决,主要是torch和torchvision版本的问题。
解决方法
把torch和torchvision都移除,
后在anaconda的环境里输入:
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
如果对你有用,就留下个赞吧!!
版权声明:本文为slh_monday原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。