Ubuntu18.04上安裝N卡驅動、CUDA、CUDNN三連

環境：Ubuntu18.04

顯卡驅動真的挺方便的，CUDA和CUDNN仍是踩了一些坑2333

 

一、安裝顯卡驅動
安裝ubuntu更新或sudo apt-get update & sudo apt-get upgrade

sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update ubuntu-drivers devices sudo ubuntu-drivers autoinstall

安裝顯卡驅動

sudo reboot nvidia-smi

若輸出顯卡信息則安裝成功

很是方便了

 

二、安裝CUDA
在官網https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal下載安裝包，默認下載到Downloads目錄下

cd ~/Downloads/

運行下載的.run文件，如：cuda_10.1.105_418.39_linux.run

sudo sh cuda_*_linux.run

在過程當中，選擇須要安裝的內容時，請將驅動driver去掉，不要安裝顯卡驅動！

 

配置環境變量（該步驟彷佛並不須要，沒有進行該操做也成功運行了後續測試，有待考證）

export CUDA_HOME=/usr/local/cuda export PATH=$PATH:$CUDA_HOME/bin export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} source ~/.bashrc sudo reboot

 

 

測試是否安裝成功，這裏若是沒有重啓過，貌似運行的時候會有報錯

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

查看CUDA版本

cat /usr/local/cuda/version.txt

 

三、安裝CUDNN
在官網https://developer.nvidia.com/cudnn註冊帳號並下載，選擇下載cuDNN Library for Linux，不過說實話如今官網給的版本好多好多看花了眼23333，默認下載到Downloads目錄下，解壓並複製

cd ~/Downloads/
tar -xzvf cudnn-*
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 /usr/local/cuda/lib64/libcudnn*

查看CUDNN版本

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