ubuntu16.04安裝cuda、cudnn

以cuda_9.0.176.384.81_linux.run爲例。

1.安裝cuda

 下載cuda：https://developer.nvidia.com/cuda-toolkit-archive

  注意cuda對應的驅動版本號，電腦上安裝的驅動版本必須高於此版本，如cuda_9.0.176.384.81_linux.run，對應可使用的驅動版本號要高於984.81。

   進入cuda.run所在目錄，修改程序權限

   sudo chmod 777 cuda_9.0.176.384.81_linux.run

   安裝：sudo ./cuda_9.0.176.384.81_linux.run

  

   事前安裝過驅動，所以不安裝。

 

 

  

  

添加環境變量：

sudo gedit ~/.bashrc 

末尾添加

exportPATH=/usr/local/cuda-9.0/bin:$PATH

exportLD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

 

使設置生效：source ~/.bashrc

 

驗證cuda是否安裝成功

 

cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

 

2.安裝cudnn

    對應cuda版本下載cuDNN：https://developer.nvidia.com/rdp/cudnn-download

    解壓cudnn包（以7.6爲例）

tar -zxvf cudnn-9.2-linux-x64-v7.6.1.34.tgz

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

  驗證安裝：

 

nvcc -V

或者查看cuDNN版本驗證：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2