如何安装Pytorch

关注公众号：『AI学习星球』
算法学习、4对1辅导、论文辅导或核心期刊可以通过公众号或CSDN滴滴我

安装torch时候需要和torchvision有版本对应关系，故做笔记方便后期查看

• 错误分析：
  • 安装pytorch或torchvision时，无法找到对应版本
  • cuda可以找到，但是无法转为.cuda()
• 以上两种或类似错误，一般由两个原因可供分析：
  • cuda版本不合适，重新安装cuda和cudnn
  • pytorch和torchvision版本没对应上

torch和torchvision版本对应关系

安装torch和torchvision时，两个包都有版本对应关系。

torch	torchvision	python	cuda
2.2	0.17	>=3.8, <=3.11	11.8, 12.1
2.1.0	0.16	>=3.8, <=3.11	11.8, 12.1
2.0.0	>0.14	>=3.7, <=3.10	11.7, 11.8
1.13.0	0.14	>=3.7, <=3.11	11.6, 11.7
1.12.0	0.13	>=3.7, <=3.10	10.2, 11.3, 11.6
1.11.0	0.12.0	>=3.7, <=3.10	10.2, 11.3
1.10.2	0.11.3	>=3.6, <=3.9	10.2, 113
1.10.1	0.11.2	>=3.6, <=3.9	10.2, 11.3
1.10.0	0.11.1	>=3.6, <=3.9	20.2, 11.3
1.9.1	0.10.1	>=3.6, <=3.9	10.2, 11.3
1.9.0	0.10.0	>=3.6, <=3.9	10.2, 11.3
1.8.0	0.9.0	>=3.6, <=3.9	10.2, 11.1
1.7.1	0.8.2	>=3.6, <=3.9	9.2, 10.1, 10.2， 11.0
1.7.0	0.8.0	>=3.6, <=3.8	9.2, 10.1, 10.2， 11.0
1.6.0	0.7.0	>=3.6, <=3.8	9.2, 10.1, 10.2
1.5.1	0.6.1	>=3.5, <=3.8	9.2, 10.1, 10.2
1.5.0	0.6.0	>=3.5, <=3.8	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

下载链接

不管是cpu的还是gpu的都在这里进行下载

在选择下载的时候，可根据自己的需要来选择

1. cpu的包还是gpu的包（这里选择的是cpu）
2. torch的包的版本是多少（这里选择的是1.4.0）
3. 是python几（这里选择的是python3.7）
4. 是哪个系统的包（这里是win系统）
5. 下载好torch也要下载对应的torchvision（根据表，torchvision要0.5.0的包）

• torch的包
  

• torchvision的包
  

导入离线包

在下载好后，通过导入离线包的方式进行导入。在导入前，建议大家导入到虚拟环境里

这里我找个导入离线包的demo，大家可以作为参考（这个是其他包，而不是torch包）

1. 比如pyspark-2.2.0.post0.tar.gz下载到了D:\下载\迅雷下载
   

2. 导入离线包，pip install 包名.whl
   

当提示Successfully installed ....就说明安装好了。

验证是否安装成功

新建python文件或者打开交互式界面，输入代码

import torch
torch.__version__
==> 2.0.0
torch.cuda.is_available()
==> True

关注公众号：『AI学习星球』
算法学习、4对1辅导、论文辅导或核心期刊可以通过公众号或CSDN滴滴我