CUDA 查看驅動兼容性:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.htmlhtml
查看GCC 與CUDA 驅動版本的兼容性linux
安裝CUDNbash
這裏選擇的是cudn9.0測試
sudo chmod a+x cuda_9.0.xxx_384.116.xx_linux.run #更改權限
sudo ./cuda_9.0.xxx_384.116.xx_linux.run網站
(注意:執行後會先出現一個聲明,須要閱讀到100%,贊成聲明後纔會開始安裝。)
安裝時首先會有一系列提示讓你確認,可是注意,有個讓你選擇是否安裝nvidia361驅動時,必定要選擇否:ui
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.85?spa
accept3d
ncode
yhtm
y
y
由於前面咱們已經安裝了更加新的nvidia384,因此這裏不要選擇安裝。其他的都直接默認或者選擇是便可。
可能出現的錯誤: (我沒遇到)
安裝cuda時可能有下面的信息
Installing the CUDA Toolkit in /usr/local/cuda-8.0 …
Missing recommended library: libGLU.so
Missing recommended library: libX11.so
Missing recommended library: libXi.so
Missing recommended library: libXmu.so
緣由是缺乏相關的依賴庫,安裝相應庫就解決了:
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
環境變量配置
打開~/.bashrc文件:
sudo gedit ~/.bashrc
將如下內容寫入到~/.bashrc尾部:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存關閉後,
source ~/.bashrc
cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery sudo make sudo ./deviceQuery
若是顯示的是關於GPU的信息,則說明安裝成功了
cuDNN 版本
安裝cudnn
進入解壓後的cudnn-9.0-linux-x64-v7.0.tgz文件cuda,在終端執行下面的指令安裝:
tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
cd cuda
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
而後更新軟鏈接:
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.7.0.5 # 本身查看.so的版本
sudo ln -sf libcudnn.so.7.0.5. libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
查看cudnn版本
在終端輸入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
便可看到相關版本