赞
踩
最近安装torchvision时总是失败,提示torchvision版本和torch版本不匹配,通过技术交流群里面大神指点,发现torchvision版本和torch版本有匹配关系,现将采坑经验分享如下:
①torchvision和pytorch版本对应关系说明地址:https://github.com/pytorch/vision
② torchvision下载地址:https://download.pytorch.org/whl/cu110/torch_stable.html
pytorch安装本文不再赘述,如果想了解可以看我另外一篇博客:https://blog.csdn.net/weixin_41267342/article/details/112066981,本文只讲述torchvision安装。
1、查看自己pytorch版本方法:
import torch print(torch.__version__)
打印结果:1.7.1+cu110,pytorch版本为1.7.1,cu110表示支持gpu加速运算,gpu版本为:11
2、网上查资料,安装touchvision方式如下:
①Anaconda:
conda install torchvision -c pytorch
②pip:
pip install torchvision
③From source:
python setup.py install # or, for OSX # MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install
我在pycharm的terminal中执行命令:pip install torchvision
但在安装过程中,会报错:
ERROR: Could not find a version that satisfies the requirement torch==1.4.0 (from torchvision) (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2, 1.7.1)
ERROR: No matching distribution found for torch==1.4.0 (from torchvision)
该错误信息即为torchvision版本和torch版本不匹配。
3、然后接下来就是想办法找和pytorch匹配的torvision版本了。
网上查资料,很多给的版本都在1.5一下,而我的pytorch版本为1.7.1+cu110,超了文章最新中版本。
然后就是想办法找最新的版本匹配说明,这个在github上有,具体地址为:
https://github.com/pytorch/vision
可以看到最新更新的官网说明:
我的1.7.1匹配的torchvison版本为0.8.2
4、下载torchvison地址:https://download.pytorch.org/whl/cu110/torch_stable.html
torchvision-0.8.2%2Bcu110-cp38-cp38-win_amd64.whl中的命名解释:
① torchvision-0.8.2:表示 torchvision版本为:0.8.2
③cp38:指的是anaconda中的python版本为python38
我的anacond中python版本为python38,在windows64位系统上运行,故需要的版本为:torchvision-0.8.2%2Bcu110-cp38-cp38-win_amd64.whl
5、下载后放到指定位置,例如:F:\WorkSpace\python\project_one>,通过pycharm的terminal窗口执行pip install torchvision-0.8.2+cu110-cp38-cp38-win_amd64.whl
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。