PyToch和Torchvision版本对应及快速安装_怎么查看pytorch跟torchvision对应关系

作者：正经夜光杯 | 2024-08-13 14:03:20

踩

怎么查看pytorch跟torchvision对应关系

在安装Pytorch(GPU或CPU版)时，如果TorchVision版本对应不上，有可能导致当前代码脚本运行失败。正确安装最好的办法是到Nvidia官网去安装要求安装。还有一个方法就是直接去网站下载TorchVision的wheel文件（https://download.pytorch.org/whl/torch_stable.html）进行安装，这样不仅安装速度快，而且安装TorchVision的同时会自动安装相对应的pytorch（强烈建议新手使用这个方法以免安装出错），wheel文件安装方法也很简单，直接到下载好的wheel文件目录下（Windows和Linux都是一样的方法）输入：pip install filename 就可以了。

以下是Pytorch和TorchVision版本对应关系：

pytorch	torchvision	python	cuda
2.2.0	0.17.0	>3.7	11.8,12.1
2.1.0	0.16.0	>3.7	11.8,12.1
2.0.0	>0.14	>3.7	11.7,11.8
1.12.0	0.12	3.7-3.9	10.2,11.3,11.6
1.11.0	0.12.0	>=3.6	11.3,10.2
1.10.0/1	0.11.0/2	>=3.6	10.2,11.3
1.9.0	0.10.0	>=3.6	10.2,11.3
1.8.0	0.9.0	>=3.6	10.2,11.1
1.7.1	0.8.2	>=3.6	9.2, 10.1,10.2,11.0
1.7.0	0.8.0	>=3.6	9.2, 10.1,10.2,11.0
1.6.0	0.7.0	>=3.6	9.2, 10.1,10.2
1.5.1	0.6.1	>=3.6	9.2, 10.1,10.2
1.5.0	0.6.0	>=3.6	9.2, 10.1,10.2
1.4.0	0.5.0	==2.7, >=3.5, <=3.8	9.2, 10.0
1.3.1	0.4.2	==2.7, >=3.5, <=3.7	9.2, 10.0
1.3.0	0.4.1	==2.7, >=3.5, <=3.7	9.2, 10.0
1.2.0	0.4.0	==2.7, >=3.5, <=3.7	9.2, 10.0
1.1.0	0.3.0	==2.7, >=3.5, <=3.7	9.0, 10.0
<1.0.1	0.2.2	==2.7, >=3.5, <=3.7	9.0, 10.0

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/正经夜光杯/article/detail/975259