Ubuntu | 安装GPU版本的pytorch

1. 安装命令

从下面链接找到符合自己环境的安装命令

历史版本Pytorch：INSTALLING PREVIOUS VERSIONS OF PYTORCH

最新版本Pytorch安装命令

我用的是：

pip3 install torch==1.6.0+cu92 torchvision==0.7.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

有conda的也可以用

conda install pytorch torchvision torchaudio cudatoolkit=9.2 -c pytorch

2. 验证安装成功

进入python环境

import torch
torch.cuda.is_available()

如果返回True就是正常的。

3. 关于cuda的版本等问题

我第一次安装pytorch的时候，设置好了cuda环境，cuda我记得是9.1;

这次安装我用nvcc -V等查看cuda版本，但是发现没有cuda版本，好像没有安装cuda，我没管它。继续用上述命令安装的。最后也安装成功了，有点玄。

4. 加速安装

4.1 先查看cuda版本

nvidia-smi倒也能显示cuda的版本，但是我遇到的这个版本不准，用下面这个显示的版本是准的。

cat /usr/local/cuda/version.txt

4.2 根据python和cuda版本确定安装包的版本

从清华镜像站 Index of /anaconda/cloud/pytorch/linux-64/ 找到要安装的pytorch相关的包，我是下载了这几个：pytorch，torchvision，torchtext，torchaudio。

下载方法用 wget 该包的链接地址即可。

4.3 用conda安装本地包

然后安装的时候conda命令如下：

conda install --use-local torch包名称   

值得一提的是，不是前面安装了好几个包嘛，安装顺序问题不知道有没有影响，我是先装了torchvision等三个，最后才安装了torchvision。