Ubuntu安裝cuda和cudnn及測試方法

分開安裝驅(qū)動和cuda

1、安裝顯卡驅(qū)動

(1)方法一從官網(wǎng)下載驅(qū)動編譯

? ? ? ? NVIDIA的中文官網(wǎng)為nvidia顯卡驅(qū)動

1)打開終端輸入以下命令刪除舊驅(qū)動

????????sudo apt-get purge nvidia*

2)禁用自帶的 nouveau nvidia驅(qū)動

? ? ? ? 首先利用命令行打開

????????sudo gedit /etc/modprobe.d/blacklist.conf

? ? ? ? 在文件末端加入

????????blacklist nouveau ?

? ? ? ? options nouveau modeset=0

????????然后更新

????????sudo update-initramfs -u

????????最后重啟

????????終端下輸入命令 lsmod | grep nouveau 查看是否被禁用


沒有信息輸出說明禁用成功

3)下載好對應(yīng)gpu版本.run 形式的顯卡驅(qū)動

? ? ? ? ?圖形化界面按住ctrl+alt+(f1~f6的其中一個)進(jìn)入命令行模式輸入用戶名和密碼,然后在輸入

? ? ? ? ?sudo service lightdm?stop?

4)給驅(qū)動run文件賦予執(zhí)行權(quán)限:

? ? ? ? ?sudo chmod ?a+x NVIDIA-Linux-x86_64-396.18.run

? ? ? ? ?然后開始安裝

? ? ? ? ?sudo ./NVIDIA-Linux-x86_64-396.18.run?

5)掛載顯卡驅(qū)動

? ? ? ? modprobe nvidia

6)檢查是否安裝成功

? ? ? ? ?nvidia-smi

? ? ? ? 如下圖則成功


(2)添加官方的ppa源

? ?1)快捷鍵ctrl+alt+T打開命令終端輸入

? ? ? ? ? ? ?sudo?add-apt-repository ppa:graphics-drivers/ppa

? ?2)然后刷新軟件庫并安裝最新驅(qū)動

? ? ? ? ? ? sudo apt-get update??

? ?3)使用終端命令查看Ubuntu推薦的驅(qū)動版本

? ? ? ? ? ? ubuntu-drivers devices


本機(jī)推薦安裝415,430,390從中選擇一個安裝

4)采用apt-get命令在終端安裝

? ? ? ? sudo apt-get install nvidia-430 nvidia-settings nvidia-prime

5)重啟系統(tǒng)并驗證 ?打開終端并輸入

? ? ? ? nvidia-smi


二、安裝cuda和cudnn

cuda的官方網(wǎng)址cudnn的官方網(wǎng)址,cudnn的下載需要注冊賬號。

(1)安裝cuda

????1)驗證cuda的安裝條件

????首先,在終端輸入 lspci | grep -I nvidia 查看你的gpu版本信息


? ? ? 接著,驗證是否安裝gcc,終端輸入?gcc –version ,如果未安裝,則輸入sudo?apt-get? install??build-essential進(jìn)行安裝


????最后,驗證系統(tǒng)是否安裝了kernel header和 package development

????終端輸入?uname –r 查看內(nèi)核 接著輸入sudo apt-get install linux-headers-$(uname -r)

? ? 即? ?sudo apt-get install linux-headers-4.18.20-25-generic

? ? 就可以安裝對應(yīng)kernel版本的kernel header和package development


2)安裝cuda

????終端輸入 sudo sh 你下載的cuda.run

????然后一直按住回車鍵,直到服務(wù)條款顯示到100%。除了driver選項選擇no其他都選擇yes。

3)設(shè)置環(huán)境變量

????在終端輸入sudo gedit ~/.bashrc

????在文本的最后輸入

????export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64

????export PATH=$PATH:/usr/local/cuda-9.0/bin

????export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

????目錄要對應(yīng)自己的cuda版本

4)然后source ?~/.bashrc 激活環(huán)境

5)在終端輸入nvcc -V出現(xiàn)下面畫面則安裝成功


(2)安裝cudnn

????1)下載對應(yīng)你cuda版本的cudnn

????2)安裝cudnn

????解壓下載的文件,可以看到cuda文件夾,在當(dāng)前目錄打開終端,執(zhí)行如下命令:

????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

????出現(xiàn)以下信息則表明安裝成功


最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容