詳細(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ù)安裝。最重要的一步，安裝程序問你是否使用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

新聞名稱：Ubuntu 16.04+CUDA8.0+Theano深度學(xué)習(xí)環(huán)境搭建
本文地址：http://uogjgqi.cn/article/dhhjcoc.html