pytorch吧 关注:1,205贴子:2,252
  • 8回复贴,共1

关于2.0.0版本的torch和torchvision

只看楼主收藏回复

请问一下2.0版本的torch 对应应该安装哪个版本的torchvision啊,
在使用jupyter notebook时候发现import torchvision时候发现缺少了模块,我用的是anaconda的
cuda版本是11.7,Python版本3.9.16


IP属地:北京来自Android客户端1楼2023-04-16 19:36回复
    可以直接pip安装torvision不就能看到是哪个版本呢,然后再卸载就行了


    IP属地:安徽来自Android客户端2楼2023-04-16 20:50
    收起回复
      或许楼主可以试试pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117


      IP属地:天津来自Android客户端3楼2023-04-16 20:53
      收起回复
        官网版本对应链接https://github.com/pytorch/vision
        torch2.0以下的torchvision默认是torch的后面版本号+1,比如torch 1.12对应torchvision是 0.13.0。
        torch2.0以上的从0.15.0开始依次递进,比如torch 2.0.1对应torchvision 0.15.1。


        IP属地:广东6楼2023-12-11 23:57
        回复