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 即爲安裝成功。