一开始看到自己电脑的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/
