PyTorch GPU版安装步骤

1. 下载并安装CUDA

官网下载链接：https://developer.nvidia.com/zh-cn/cuda-toolkit

我是Windows, 然后Server 2022应该是最新版本吧我猜。选择local是因为network在安装过程中还是得下载的，索性就一次性下载好。所以local还是network大家自行选择哈~

安装包下载很快，双击就可以进入安装页面了，接下来按需要选就行了：


我是推荐啥就选啥，也不知道高级的自定义是什么意思

打勾

如果你是local就会直接安装。如果你是network就还需要下载:



到这里CUDA就安装好了！

2. 下载并安装GPU版PyTorch

官网链接 (似乎需要科学上网)：https://pytorch.org/get-started/locally/

上不去的同学没关系哈~跟我一样选择的在powershell执行以下代码

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

如果conda下载很慢的同学可以尝试添加清华镜像，指令：

conda config --set show_channel_urls yes
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 config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
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/

然后运行(记得添加清华镜像以后不要-c pytorch)：

conda install pytorch torchvision torchaudio cudatoolkit=11.3

3. 检查是否安装成功

在jupyter notebook中尝试import torch，如果没报错说明安装成功~

4. 2021.12.4更新

打开notebook拟合RNN的时候报了一个NotImplementError，经过一番搜索我发现是因为还没有配置完，可以参照：https://www.tensorflow.org/install/source_windows进行继续配置！