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