pytorch支持的python版本 pytorchcuda版本

1：登录Pytorch官网

官网地址：https://pytorch.org/

2：选择Pytorch版本信息

点击install，进入安装页面

这里我选择的是Pytorch 1.9.0版，CUDA版本号为11.1

如何查看CUDA的版本信息呢？
（1）首先确定你的显卡支持CUDA，这个可以百度

这里我的显卡是n卡GTX 1650

win+R打开cmd命令行，输入命令nvidia-smi

确保显卡驱动版本大于400才可，我的CUDA版本为11.1（2）若显卡不支持CUDA，则选择cpu版本即可

3：下载并安装Pytorch包

复制安装页面上的命令

打开命令行进入Python的虚拟环境，粘贴上面的命令，于是开始了最烦的一步，因为这个官网服务器在海外，国内大陆下载速度很慢不说，还经常下到一般断开，因为命令行中下载不支持断点续连，所以一旦下载失败，我连Anaconda都要重新装一遍，于是开始了漫长的下载-重装-下载-重装，耗费我一下午时间还没有装好，用清华、中科大等镜像也无法解决问题。于是我尝试用离线安装包在本地安装：

（1）在Pytorch官网安装页面上改为pip安装

注意图中红线位置的版本号，我的为：torch==1.9.0+cu111 torchvisinotallow==0.10.0+cu111；

然后去离线下载网址下载对应的torch和torchvision安装包，推荐用迅雷下载速度快。

下载完成后如图：

（2）打开命令行，创建虚拟环境

（3）进入环境

（4）输入命令：pip install (whl文件保存路径)\文件名称

分别安装两个文件

注意：还是需要联网安装，因为安装包可能会有更新，会下载一部分更新后的包！！！（5）检验安装是否成功

进入python 控制台
import torch
无报错
torch.cuda.is_available()
返回True
安装成功！！！