ubuntu18.04安装CUDA10.1+cudnn

虽然安装了好多次，但有时候仍然需要搜索一下教程，所以自己记录下安装过程，避免踩坑。

 

1.安装显卡驱动

采用最简单的方法，可以避免其他方法出现在登陆界面循环出不来，直接在software&updates->additional Drivers里面选择驱动版本更新（如果没有，需要将下载下载源换乘国内镜像，推荐清华 参考：https://blog.csdn.net/Justdoforever/article/details/90949841）然后Revert

然后打开终端

nvidia-smi

出现下面界面则成功

 

2.安装CUDA(CUDA各种版本)

下载后执行命令

sudo sh cuda_10.1.105_418.39_linux.run

后续如果有安装驱动的选项，选择不安装

安装完成后添加环境变量

sudo gedit /etc/bash.bashrc

复制下列内容

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

然后重新初始化执行该文件使之生效

source /etc/bash.bashrc
sudo ldconfig

重开终端，执行

nvcc -V

出现cuda版本即安装成功。

 

3.安装cudnn（下载地址）

需要注册才能下载，选择与CUDA对应的cudnn版本 且选择cuDNN Library for Linux（是一个压缩包）

下载好后解压,解压后就是cuda文件夹，将对应文件复制到 cuda目录下

sudo cp cuda/include/cudnn.h    /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn*    /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

--over