ubuntu 安装 CUDA、 cuDNN 的tips
?
CUDA 查看驅動兼容性:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
查看GCC 與CUDA 驅動版本的兼容性
?
安裝CUDN
這里選擇的是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驅動時,一定要選擇否:
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.85?
accept
n
y
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
測試CUDA的samples
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
即可看到相關版本
?
?相關網站:?https://developer.nvidia.com/rdp/cudnn-archive
? ? ? ? ? ? ? ? ? ??https://www.tensorflow.org/install/source?hl=zh-cn#linux
?
?
?
轉載于:https://www.cnblogs.com/billwong/p/10884099.html
總結
以上是生活随笔為你收集整理的ubuntu 安装 CUDA、 cuDNN 的tips的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: C/C++调试:gdbserver的简单
- 下一篇: LeetCode 705. Design