深度學習環境搭建(二)—— Ubutun16.04 安裝CUDA、CUDNN

 

        該篇關於安裝CUDA、CUDNN的博客是基於已經安裝好NVIDA驅動的基礎,具體如何安裝NVIDIA驅動可參考我的上一篇博客

一、安裝CUDA

(1)CUDA官網選擇適合自己系統的版本下載。(建議選擇CUDA10.0或CUDA10.1,雖說還有很多人在用CUDA9但是CUDA10的趨勢越來越主流,很多product都是基於CUDA10運行的)

我的系統是Ubuntu16.04、64位,選擇CUDA10版本

(2)安裝CUDA(如果是參考我的上一篇博客安裝NVIDIA驅動,那麼一定採用runfile的方式安裝CUDA)

sudo sh cuda_10.0.130_430.64_linux.run

(3)一定要選擇不安裝驅動,選擇如下:(或者不勾選CUDA driver)

會提示:

***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 384.00 is required for CUDA 10.0 functionality to work.

To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:

sudo <CudaInstaller>.run -silent -driver

這是因爲我們在安裝CUDA的時候沒有選擇安裝驅動,提示需要安裝驅動,忽略就行。

(4)添加環境變量(安裝CUDA時,會自動在/usr/local目錄下生成cuda的文件目錄)

export CUDA_HOME=/usr/local/cuda 
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

(5)保存並退出

source ~/.bashrc

(6)測試是否成功

上面的選項選擇安裝了CUDA例子,運行其中一個來測試是否安裝成功:

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

結果如下就表明成功啦

二、安裝CUDNN

(1)在官網下載安裝包,需要註冊登錄才能下載。選擇適合自己的版本

(2)下載完成後解壓並進入文件夾:

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 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

(3)在終端查看CUDNN版本:

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

Yes, you have completed the complicated work.

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章