新聞中心
準(zhǔn)備工作

1.查看GPU是否支持CUDA
lspci | grep -i NVIDIA
2.查看Linux版本
uname -m && cat /etc/*release
nvidia驅(qū)動(dòng)
1. 先卸載原有N卡驅(qū)動(dòng)
#for case1: original driver installed by apt-get:
sudo apt-get remove –purge nvidia*
#for case2: original driver installed by runfile:
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-384.59.run –uninstall
2. 禁用nouveau驅(qū)動(dòng)
sudo gedit /etc/modprobe.d/blacklist.conf
在文本最后添加:
blacklist nouveau
options nouveau modeset=0
然后執(zhí)行:
sudo update-initramfs -u
重啟之后,可以查看nouveau有沒有運(yùn)行:
lsmod | grep nouveau # 沒輸出代表禁用生效
3.禁用X-Window服務(wù)
sudo service lightdm stop #這會(huì)關(guān)閉圖形界面
按Ctrl-Alt+F1進(jìn)入命令行界面,輸入用戶名和密碼登錄。
4.命令行安裝驅(qū)動(dòng)
驅(qū)動(dòng)網(wǎng)址https://www.nvidia.cn/Download/index.aspx?lang=cn
#給驅(qū)動(dòng)run文件賦予執(zhí)行權(quán)限:
sudo chmod +x NVIDIA-Linux-x86_64-384.59.run
#后面的參數(shù)非常重要,不可省略:
sudo ./NVIDIA-Linux-x86_64-384.59.run –no-opengl-files
5.Driver測(cè)試:
nvidia-smi #若列出GPU的信息列表,表示驅(qū)動(dòng)安裝成功
nvidia-settings #若彈出設(shè)置對(duì)話框,亦表示驅(qū)動(dòng)安裝成功
CUDA安裝
1.下載對(duì)應(yīng)系統(tǒng)的CUDA
網(wǎng)址http://developer.nvidia.com/cuda-downloads 選擇runfile安裝
2.安裝
sudo sh cuda_
開始安裝之后,需要閱讀說(shuō)明,可以使用Ctrl + C直接閱讀完成,或者使用空格鍵慢慢閱讀。下面為安裝選項(xiàng):
(是否同意條款,必須同意才能繼續(xù)安裝)
accept/decline/quit: accept
(這里不要安裝驅(qū)動(dòng),因?yàn)橐呀?jīng)安裝最新的驅(qū)動(dòng)了,否則可能會(huì)安裝舊版本的顯卡驅(qū)動(dòng),導(dǎo)致重復(fù)登錄的情況)
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: n
Install the CUDA 10.0 Toolkit?(是否安裝CUDA 10 ,這里必須要安裝)
(y)es/(n)o/(q)uit: y
Enter Toolkit Location(安裝路徑,使用默認(rèn),直接回車就行)
[ default is /usr/local/cuda-10.0 ]:
Do you want to install a symbolic link at /usr/local/cuda?(同意創(chuàng)建軟鏈接)
(y)es/(n)o/(q)uit: y
Install the CUDA 10.0 Samples?(不用安裝測(cè)試,本身就有了)
(y)es/(n)o/(q)uit: n
Installing the CUDA Toolkit in /usr/local/cuda-10.0 …(開始安裝)
3.配置環(huán)境變量
sudo gedit ~/.bashrc
末尾加入
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
更新
source ~/.bashrc
4.驗(yàn)證安裝
查看cuda版本
nvcc -V
CUDA Sample測(cè)試:
#編譯并測(cè)試設(shè)備 deviceQuery:
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
make
./deviceQuery
#編譯并測(cè)試帶寬 bandwidthTest:
cd ../bandwidthTest
make
./bandwidthTest
如果這兩個(gè)測(cè)試的最后結(jié)果都是Result = PASS,說(shuō)明CUDA安裝成功。
*5.CUDA卸載
在命令行中輸入
sudo apt-get remove cuda
sudo apt-get autoclean
sudo apt-get remove cuda*
然后在目錄切換到usr/local/下
cd /usr/local/
sudo rm -r cuda-9.1
cuDNN安裝
下載對(duì)應(yīng)版本cuDNN https://developer.nvidia.com/cudnn
tar xvzf cudnn-9.2-linux-x64-v7.1
sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo ldconfig
Reference
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
總結(jié)
以上所述是小編給大家介紹的Ubuntu16.04安裝nvidia驅(qū)動(dòng)+CUDA+cuDNN的教程,大家如有疑問(wèn)可以留言,或者聯(lián)系站長(zhǎng)。感謝親們支持?。?!
如果你覺得本文對(duì)你有幫助,歡迎轉(zhuǎn)載,煩請(qǐng)注明出處,謝謝!
成都服務(wù)器租用選創(chuàng)新互聯(lián),先試用再開通。
創(chuàng)新互聯(lián)(www.cdcxhl.com)提供簡(jiǎn)單好用,價(jià)格厚道的香港/美國(guó)云服務(wù)器和獨(dú)立服務(wù)器。物理服務(wù)器托管租用:四川成都、綿陽(yáng)、重慶、貴陽(yáng)機(jī)房服務(wù)器托管租用。
網(wǎng)站欄目:詳解Ubuntu16.04安裝nvidia驅(qū)動(dòng)+CUDA+cuDNN的教程
鏈接地址:http://www.dlmjj.cn/article/djcogih.html


咨詢
建站咨詢
