linux安装pytorch（包括cuda与cudnn）

安装cuda

先查看显卡信息，有显卡才能装cuda。

lspci | grep -i nvidia

从官网下载cuda，特别注意：要选择run，不要选择deb。
然后把cuda上传到服务器。

$ chmod +x cuda_10.1.105_418.39_linux.run
$ sudo ./cuda_10.1.105_418.39_linux.run

然后会弹出窗口，输入accept，键盘方向键选中driver，按空格键取消驱动的选择。
安装完成后添加环境变量

export PATH=/usr/local/cuda-10.1/bin:$PATH

cuda安装后的检查一下

nvcc -V

cudnn的安装

根据cuda版本选择对应的cudnn进行下载，下载完成后上传到服务器。

然后解压cudnn文件，并把cudnn文件夹内的文件拷贝到cuda安装目录即可。

tar -xvf cudnn-10.1-linux-x64-v7.6.4.38.tgz
sudo cp cuda/include/cudnn.h    /usr/local/cuda-10.1/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64/
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h   /usr/local/cuda-10.1/lib64/libcudnn*

安装pytorch

在官网选择版本与系统，然后按照官网给出的命令安装即可。

如果直接执行命令下载较慢，可以用清华源，conda的清华源添加方式可以百度，这里提供linux的添加。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

添加清华源后安装pytorch过程中出错，尝试一下重新执行安装pytorch的命令即可。

验证

以上安装完成后，打开python，执行以下脚本。

import torch 

print(torch.cuda.is_available())

若输出true，则表示cuda可以使用了。然后开始愉快的torch之旅吧。