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进行继续配置!