Linux 16.04 安裝Nvidia Driver, Cuda 10.0, Cudnn

1. 型號(hào)

(1) 顯卡 GTX 1060, 6GB顯存
(2)驅(qū)動(dòng) Nvidia 450.66
(3) Cuda 10.0
(4) CuDnn 7.6.4.38


Driver

2.安裝Driver

參考 (https://blog.csdn.net/lihe4151021/article/details/90083431)
(1)刪除原有驅(qū)動(dòng)(如有)

sudo apt-get remove --purge nvidia*
image.png

(2)禁用nouveau


查看Nouveau是否禁用,有輸出則未禁用
sudo gedit /etc/modprobe.d/blacklist.conf #打開該文件
#在文件最后添加
blacklist nouveau
options nouveau modeset=0
image.png

image.png

(3)更新并重啟系統(tǒng)

update-initramfs -u
reboot

(4) 檢查nouveau是否已禁用

lsmod | grep nouveau
查看Nouveau是否禁用

沒(méi)有信息顯示,說(shuō)明nouveau已被禁用,接下來(lái)可以安裝nvidia的顯卡驅(qū)動(dòng)
(5) 關(guān)閉圖形界面
ubuntu下按ctrl+alt+f1進(jìn)入字符界面,然后在命令行界面下輸入

    sudo service lightdm stop      //這個(gè)是關(guān)閉圖形界面,不執(zhí)行會(huì)錯(cuò)。
image.png

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

sudo chmod  a+x NVIDIA-Linux-x86_64-450.66.run
image.png

(7) 安裝

sudo ./NVIDIA-Linux-x86_64-450.66.run -no-x-check -no-nouveau-check -no-opengl-files 
 //只有禁用opengl這樣安裝才不會(huì)出現(xiàn)循環(huán)登陸的問(wèn)題

image.png

描述:
-no-x-check:安裝驅(qū)動(dòng)時(shí)關(guān)閉X服務(wù)
-no-nouveau-check:安裝驅(qū)動(dòng)時(shí)禁用nouveau
-no-opengl-files:只安裝驅(qū)動(dòng)文件,不安裝OpenGL文件
(8) 安裝過(guò)程中的選項(xiàng)
The distribution-provided pre-install script failed! Are you sure you want to continue? 選擇 yes 繼續(xù)。


image.png

Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later? 選擇 No 繼續(xù)。


image.png

image.png

image.png

問(wèn)題沒(méi)記住,選項(xiàng)是:install without signing

問(wèn)題大概是:Nvidia’s 32-bit compatibility libraries? 選擇 No 繼續(xù)。


image.png

Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up. 選擇 Yes 繼續(xù)
這些選項(xiàng)如果選擇錯(cuò)誤可能會(huì)導(dǎo)致安裝失敗,沒(méi)關(guān)系,只要前面不出錯(cuò),多嘗試幾次就好。


image.png

image.png

(9)打開圖形界面

    sudo service lightdm start      //這個(gè)是打開圖形界面

(10) 檢查是否安裝成功

nvidia-smi
image.png

3.安裝CUDA10.0

參考
(https://blog.csdn.net/lihe4151021/article/details/90237681)
(https://zhuanlan.zhihu.com/p/47330858)

(1)

cd Downloads 
image.png

(2)

sudo sh cuda_10.0.130_450.48_linux.run
image.png

開始安裝,按提示一步步操作,按住回車看完聲明。按照提示輸入相應(yīng)字符,例如有的需要輸入accept,有的需要輸入yes。
安裝成功后,會(huì)顯示installed,否則會(huì)顯示failed。


image.png

image.png

image.png

(3)

sudo gedit /etc/profile
image.png

在打開的文件末尾,復(fù)制添加以下兩行:

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
image.png

保存文件,并重啟。因?yàn)閟ource /etc/profile是臨時(shí)生效,重啟電腦才是永久生效。
到這里安裝應(yīng)該就成功了,后面是一些驗(yàn)證的工作,其實(shí)一般不用驗(yàn)證。
(4)重啟電腦,然后檢查上述的環(huán)境變量是否設(shè)置成功
a、 驗(yàn)證驅(qū)動(dòng)版本

cat /proc/driver/nvidia/version

b、 驗(yàn)證CUDA Toolkit

nvcc -V 
image.png

4.安裝CuDNN

(1)下載安裝文件
按需求下載cudnn的安裝文件:https://developer.nvidia.com/rdp/cudnn-archive

(2)安裝cudnn
我們從Nvidia官網(wǎng)上下載下來(lái)的cudnn for linux的文件格式是.solitairetheme8,想要解壓的話需要先轉(zhuǎn)成tgz格式再解壓:


tar -xvf cudnn-10.0-linux-x64-v7.3.1.20.tgz

解壓下載的文件,可以看到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*

image.png

(3)查看cudnn版本
在終端輸入

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

如果出現(xiàn)下圖所示版本信息,說(shuō)明安裝成功。


image
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

推薦閱讀更多精彩內(nèi)容