1. 型號(hào)
(1) 顯卡 GTX 1060, 6GB顯存
(2)驅(qū)動(dòng) Nvidia 450.66
(3) Cuda 10.0
(4) CuDnn 7.6.4.38
2.安裝Driver
參考 (https://blog.csdn.net/lihe4151021/article/details/90083431)
(1)刪除原有驅(qū)動(dòng)(如有)
sudo apt-get remove --purge nvidia*
(2)禁用nouveau
sudo gedit /etc/modprobe.d/blacklist.conf #打開該文件
#在文件最后添加
blacklist nouveau
options nouveau modeset=0
(3)更新并重啟系統(tǒng)
update-initramfs -u
reboot
(4) 檢查nouveau是否已禁用
lsmod | grep 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ò)。
(6) 給驅(qū)動(dòng)run文件賦予執(zhí)行權(quán)限:
sudo chmod a+x NVIDIA-Linux-x86_64-450.66.run
(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)題
描述:
-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ù)。
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ù)。
問(wèn)題沒(méi)記住,選項(xiàng)是:install without signing
問(wèn)題大概是:Nvidia’s 32-bit compatibility libraries? 選擇 No 繼續(xù)。
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ò),多嘗試幾次就好。
(9)打開圖形界面
sudo service lightdm start //這個(gè)是打開圖形界面
(10) 檢查是否安裝成功
nvidia-smi
3.安裝CUDA10.0
參考
(https://blog.csdn.net/lihe4151021/article/details/90237681)
(https://zhuanlan.zhihu.com/p/47330858)
(1)
cd Downloads
(2)
sudo sh cuda_10.0.130_450.48_linux.run
開始安裝,按提示一步步操作,按住回車看完聲明。按照提示輸入相應(yīng)字符,例如有的需要輸入accept,有的需要輸入yes。
安裝成功后,會(huì)顯示installed,否則會(huì)顯示failed。
(3)
sudo gedit /etc/profile
在打開的文件末尾,復(fù)制添加以下兩行:
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存文件,并重啟。因?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
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*
(3)查看cudnn版本
在終端輸入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出現(xiàn)下圖所示版本信息,說(shuō)明安裝成功。