当前位置:   article > 正文

pythorch版本和torchvision版本对应关系及torchvision安装_torch2.0.0对应的torchvision

torch2.0.0对应的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

 

 

 

声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号