Ubuntu 18.04安装Cuda和Cudnn

Ubuntu 18.04安装Cuda和Cudnn

软件配置

1. Ubuntu 18.04 LTS
2. Cuda 10.1
3. Cudnn 7

step 1: 安装Nvidia显卡驱动

1. 使用 ubuntu-drivers devices 查看， 安装界面中推荐的驱动版本，例如：driver : nvidia-driver-440 - third-party free recommanded.

2. 卸载旧的Nvidia驱动

   sudo apt remove --purge nvidia*
   

3. 更新显卡驱动软件源

   sudo apt update
   sudo apt upgrade
   sudo add-apt-repository ppa:graphics-drivers
   

4. 安装推荐的显卡驱动

   sudo apt install nvidia-driver-440
   

5. 测试: 终端输入nvidia-smi，显示显卡信息即为安装成功。

step 2: 安装 Cuda 10.1

1. 官网Nvidia 下载 cuda_10.1.168_418.67_linux.run 文件。
   

2. 安装Cuda.

   sudo sh cuda_10.1.168_418.67_linux.run
   

   安装过程中会出现一个选择界面，第一项显卡驱动不选，之后默认安装即可。
   .

3. 安装成功
   

4. 环境变量设置

   sudo vim ~/.bashrc
   export PATH="/usr/local/cuda-10.1/bin:$PATH"
   export LD_LIBRARY_PATH="/usr/lcoal/cud10.1/lib64:$LD_LIBRARY_PATH"
   source ~/.bashrc
   

5. 测试安装是否成功

   cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery
   sudo make
   ./deviceQuery
   

   出现Result = Pass则表示安装成功
   

Step 3: 安装cuDNN

1. 下载cuDNN v7.6.1

2. 解压cuDNN到相应的目录，运行下面命令：

   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 
   sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
   

3. 下载 libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb, libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb, libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64.deb.

   sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
   sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb
   sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64.deb
   

4. 检测是否安装成功

   cp -r /usr/local/cudnn_samples_v7 ~/
   cd ~/cudnn_samples_v7
   make clean && make
   ./mnistCUDNN
   

   出现 Test passed 即为安装成功。