Ubuntu16.04 安装pytorch cuda cudnn

1.cuda的安装

到官网下载你需要的版本

cuda11

https://developer.nvidia.com/cuda-toolkit

cuda10.2

https://developer.nvidia.com/cuda-10.2-download-archive

然后按照对应的命令就可以安装cuda。

$sudo gedit ~/.bashrc

 编辑配置文件，在末尾写入

export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH

 运行

$source ~/.bashrc

nvcc -V如果显示cuda型号就没有问他

2.cudnn

cudnn需要注册下帐号

https://developer.nvidia.com/cudnn

点击Download cuDNN，登录帐号

点击10.2这个版本，跳转到新的页面后同意协议

下载对应的三个文件

#使用命令
$sudo dpkg -i /home/czx/下载/libcudnn8_8.0.2.39-1+cuda10.2_amd64.deb 

 如果报错

dpkg-deb：错误：/home/czx/下载/libcudnn8_8.0.2.39-1+cuda10.2_amd64.deb 并非 debian 格式的包文件

这个错误是包没下好，重新下载对应的包，再次安装

$sudo dpkg -i /home/czx/下载/libcudnn8_8.0.2.39-1+cuda10.2_amd64.deb
$sudo dpkg -i /home/czx/下载/libcudnn8-dev_8.0.2.39-1+cuda10.2_amd64.deb
$sudo dpkg -i '/home/czx/下载/libcudnn8-doc_8.0.2.39-1+cuda10.2_amd64.deb'

 测试一下

$cp -r /usr/src/cudnn_samples_v8/ $HOME
$ cd  $HOME/cudnn_samples_v8/mnistCUDNN
$ make clean && make
$ ./mnistCUDNN

• Test passed!  #出现这个就是彻底成功啦！

报错：cuda failure (system has unsupported display driver / cuda driver combination) in error_util.h:91

cuda与显卡驱动不匹配，确认成功安装cuda后重启就解决啦。

 

cudnn就安装好了

3.pytorcn

https://pytorch.org/

进入官网

得到对应的下载代码(pip install torch torchvision)，如果下载速度慢记得修改源，pip下载源修改如下(在虚拟环境下操作！！！！)

$ mkdir ~/.pip
$ sudo gedit ~/.pip/pip.conf

在跳出的页面中写入源

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host=mirrors.aliyun.com

运行pip install torch torchvision（看上图你在pytorch官网得到的下载代码）完毕