建站
咨詢
售后
建站咨詢
新聞中心
詳細(xì)分享在ubuntu 16.04下搭建CUDA8.0+Theano深度學(xué)習(xí)環(huán)境的步驟，希望對(duì)大家有所幫助。
一、安裝nvidia驅(qū)動(dòng)
1、首先去官網(wǎng)上查看適合你GPU的驅(qū)動(dòng)（http://www.nvidia.com/Download/index.aspx?lang=en-us），根據(jù)自己買的型號(hào)選擇下載，放到/home/isi/cuda目錄下面，我下載的是NVIDIA-Linux-x86_64-375.39.run
2、屏蔽開源驅(qū)動(dòng) nouveau（原文章說(shuō)“可以不編輯blacklist.conf，NVIDIA.run驅(qū)動(dòng)程序可以自己停止其他開源驅(qū)動(dòng)，但是需要重啟”)
sudo gedit /etc/modprobe.d/blacklist.conf
在文件末尾添加如下內(nèi)容
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
（這里有一行空格）
3、刪除舊NVIDIA驅(qū)動(dòng)
sudo apt-get --purge remove nvidia-*(需要清除干凈)
sudo apt-get --purge remove xserver-xorg-video-nouveau
4、重啟電腦
sudo reboot
5、按Ctrl + Alt +F1(F1~F6均可)到x-server(tty), Ctrl+Alt+F7是返回
6、關(guān)閉圖形環(huán)境，否則驅(qū)動(dòng)無(wú)法正常安裝
輸入用戶名密碼登錄，然后鍵入命令后執(zhí)行（數(shù)字不能用小鍵盤輸入）
sudo service lightdm stop
成功關(guān)閉服務(wù)后將出現(xiàn)：lightdm stop/waiting
7、進(jìn)入驅(qū)動(dòng)所在的文件夾，這里就是/home/isi/cuda，執(zhí)行
sudo sh NVIDIA*.run
后面就一路Accept就可以~報(bào)錯(cuò)The distribution-provided pre-install script failed!不必理會(huì)，繼續(xù)安裝。最重要的一步，安裝程序問(wèn)你是否使用nv的xconfig文件，這里一點(diǎn)要選yes，否則在啟動(dòng)x-window時(shí)不會(huì)使用nv驅(qū)動(dòng)。
8、重啟 X-window 服務(wù)，并重啟電腦
sudo service lightdm start
sudo reboot
9、查看顯卡是否裝好
方法1：運(yùn)行下邊代碼，假如顯示"direct rendering: Yes"，則已安裝
glxinfo | grep rendering
方法2：運(yùn)行下邊代碼，出現(xiàn)GPU列表，則說(shuō)明驅(qū)動(dòng)安裝成功了
nvidia-smi
二、安裝cuda
1、從這里下載cuda的安裝文件
https://developer.nvidia.com/cuda-release-candidate-download
注意這里下載的是cuda8.0的runfile（local）文件，我放在了/home/isi/cuda文件夾下
2、進(jìn)入到/home/isi/cuda文件夾，進(jìn)行安裝
cd /home/isi/cuda
sudo sh cuda_8.0.27_linux.run
執(zhí)行后會(huì)有一系列提示讓你確認(rèn)，但是注意，有個(gè)讓你選擇是否安裝nvidia361驅(qū)動(dòng)時(shí)，一定要選擇否，因?yàn)榍懊嫖覀円呀?jīng)安裝了更加新的nvidia367，所以這里不要選擇安裝。其余的都直接默認(rèn)或者選擇是即可。
3、安裝成功后會(huì)出現(xiàn)如下界面：
===========
= Summary =
===========
Driver:NotSelected
Toolkit: Installedin/usr/local/cuda-8.0
Samples: Installedin/home/textminer
Please make sure that
– PATH includes /usr/local/cuda-8.0/bin
– LD_LIBRARY_PATH includes /usr/local/cuda-8.0/lib64,or,add/usr/local/cuda-8.0/lib64to/etc/ld.so.confandrun ldconfigasrootTouninstall the CUDA Toolkit, run the uninstall scriptin/usr/local/cuda-8.0/bin
Please see CUDA_Installation_Guide_Linux.pdfin/usr/local/cuda-8.0/doc/pdffordetailed informationonsetting up CUDA.
***WARNING: Incomplete installation! This installation didnotinstall the CUDA Driver. A driverofversion at least361.00isrequiredforCUDA8.0functionalitytowork.Toinstall the driverusingthis installer, run the following command, replacingwiththe nameofthis run file:
sudo .run -silent -driver
Logfileis/opt/temp//cuda_install_6583.log
4、安裝完畢后，再聲明一下環(huán)境變量，并將其寫入到 ~/.bashrc 的尾部:
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
5、然后設(shè)置環(huán)境變量和動(dòng)態(tài)鏈接庫(kù)，在命令行輸入：
sudo gedit /etc/profile
在打開的文件末尾加入：
exportPATH=/usr/local/cuda/bin:$PATH
6、保存之后，創(chuàng)建鏈接文件：
sudo gedit /etc/ld.so.conf.d/cuda.conf
在打開的文件中添加如下語(yǔ)句：
/usr/local/cuda/lib64
然后執(zhí)行下列代碼使鏈接立即生效
sudo ldconfig
網(wǎng)頁(yè)名稱：Ubuntu 16.04+CUDA8.0+Theano深度學(xué)習(xí)環(huán)境搭建
文章分享：http://www.5511xx.com/article/dhhjcoc.html
