安装与电脑CUDA匹配的pytorch版本

查看是否支持以及版本号

查看windows的GPU及CUDA版本

卸载pytorch

如果按的与版本不匹配，还需要卸载了
pip uninstall torch

使用pip安装（未成功）

官网生成命令

安装pytorch

官网命令

pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

查看pip当前源：

pip config list

为了防止各种问题我们 更改一下，参考

pip --default-timeout=1000 install --index-url https://pypi.tuna.tsinghua.edu.cn/simple/ torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

使用Anaconda 安装（成功）

参考

https://blog.csdn.net/watermelon1123/article/details/88122020

添加清华源

该方案就是将官网的下载路径，改到清华镜像源的路径，先用activate激活自己之前创建的环境后，在命令行输入以下代码：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

另外为了保险起见，建议同时添加第三方conda源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

设置搜索时显示通道地址

conda config --set show_channel_urls yes

然后再去pytorch官网复制下载代码，记得删掉后面的-c pytorch,因为它表示从官网进行下载，不从清华镜像源下。

根据Python和CUDA选择对应的版本，然后官方给出提示可通过运行：

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.1 

但是这里一定要注意，去掉-c pytorch，安装的时候才会默认从清华源下载相应的包，因此这里用命令行:

查看anaconda配置

conda config --show-sources

测试环境是否安装成功

import  torch
print (torch.cuda.is_available())