pythorch版本和torchvision版本对应关系及torchvision安装

最近安装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

②cu110：支持的cuda(GPU)版本为：11

③cp38：指的是anaconda中的python版本为python38

④win_amd64：用于64位window系统。

我的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

pythorch版本和torchvision版本对应关系及torchvision安装

猜你喜欢