基于anaconda和python3.7安装pytorch1.5.0

一开始看到自己电脑的CUDA版本为11.0，以为不能支持，实际上在虚拟环境中可以安装cudatoolkit的任意版本：
1.打开anaconda prompt，建立一个新环境
conda create -n your_env_name python=3.7

2.安装cudatoolkit和对应的cudnn
conda install cudatoolkit=10.2
conda install cudnn=7.6.5

3.安装pytorch
直接去官网查找指令即可https://pytorch.org/

我这里是

再在命令框里输入指令安装就完成了。

安装完之后可以输入
torch.cuda.get_device_name(0)
查看是否支持gpu

找不到特定版本时，可以添加如下源 ：

• https://mirrors.ustc.edu.cn/anaconda/pkgs/main/

• https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/

• https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

• https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/

• https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/

• https://mirrors.sjtug.sjtu.edu.cn/anaconda/cloud/conda-forge/