Ubuntu 18.04/20.04快速安裝Nvdia驅(qū)動及其CUDA

1 首先檢查基礎(chǔ)環(huán)境是否ok
sudo apt install build-essential
(里面包含了大部分基礎(chǔ)環(huán)境包括g++,make等,如果沒有網(wǎng)絡(luò)請參考我的離線配置文章)
下載nvidia驅(qū)動,最好放到/home/user/Download下方便找
https://www.nvidia.cn/Download/index.aspx?lang=cn(選擇linux64位的下載)
下載cuda,也放到Download下,cuda需要看對應(yīng)版本
https://developer.nvidia.com/cuda-toolkit-archive
cuda和驅(qū)動對應(yīng)關(guān)系看下圖

cudnn的下載頁面和對應(yīng)關(guān)系
https://developer.nvidia.com/rdp/cudnn-archive
2 禁用nouveau并重啟(安裝Nvidia顯卡的官方驅(qū)動和系統(tǒng)自帶的nouveau驅(qū)動沖突)
sudo gedit /etc/modprobe.d/blacklist.conf
在打開文本的最后一行添加:
blacklist nouveau
options nouveau modeset=0
更新,重啟:
sudo update-initramfs -u
reboot
重啟后再次查看是否禁用nouveau
lsmod | grep nouveau
如果沒有任何輸出代表禁用ok
3 安裝顯卡驅(qū)動(和centos步驟一樣)
首先
sudo init3 ---
進(jìn)入單用戶模式
此時屏幕會進(jìn)入terminal黑屏背景
按照提示login用戶之后
cd進(jìn)入你的驅(qū)動所在目錄,以我的為例子我放在了/home/user/Download下
cd /home/user/Download
添加執(zhí)行權(quán)限
sudo?chmod?a+x NVIDIA-Linux-x86_64-xxx.run
執(zhí)行安裝
sudo ./NVIDIA-Linux-x86_64-xxx.run -no-opengl-files
這里要注意如果安裝提示x服務(wù)錯誤,類似“You appear to be running an X server; please exit X before installing.”,請在后面繼續(xù)加參數(shù)變成如下的命令再運(yùn)行
sudo?./NVIDIA-Linux-x86_64-xxx.run -no-opengl-files -no-x-check
安裝過程中
32bit兼容包選擇, 這里要注意選擇NO,不然后面就會出錯。?
X-configurtion的選擇頁面NO
安裝完成后
sudo init5??切換回桌面模式
終端輸入nvidia-smi檢查驅(qū)動是否安裝完好
4?安裝CUDA(和centos步驟一樣)
這里以CUDA11.1為例子,10.x包括后面的安裝都是一樣的,依葫蘆畫瓢即可
centos也是一樣的,就不單獨(dú)寫專欄了
首先安裝CUDA的run文件
sudo ./cuda_11.1.xxx_xxx.xx_linux.run
這里需要注意,因?yàn)槲覀円呀?jīng)安裝過nvidia顯卡驅(qū)動,所以這里展開cuda的安裝選項(xiàng)的時候,我們需要去掉顯卡驅(qū)動前面的勾勾

安裝完成之后,我們需要修改bashrc
vim?~/.bashrc
之后在最下方添加下列兩行
export PATH=$PATH:/usr/local/cuda-11.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
然后source一下
source? ~/.bashrc
你在哪個用戶下安裝的驅(qū)動和cuda,對應(yīng)的用戶下bashrc就需要添加,如果不確定請在root和你當(dāng)前賬戶下都添加一下
5?編譯SAMPLE(測試CUDA是否安裝成功)
cd /usr/local/cuda-11.1/samples/
編譯
make all - j100(內(nèi)核數(shù),可以往大的寫)
編譯完成后
cd bin/x86_64/linux/release/
?然后測試cuda是否test ok
./deviceQuery
輸出結(jié)尾看到Result=PASS則代表安裝成功,恭喜
