新聞中心
詳細(xì)分享在ubuntu 16.04下搭建CUDA8.0+Theano深度學(xué)習(xí)環(huán)境的步驟,希望對大家有所幫助。
一、安裝nvidia驅(qū)動
1、首先去官網(wǎng)上查看適合你GPU的驅(qū)動(http://www.nvidia.com/Download/index.aspx?lang=en-us),根據(jù)自己買的型號選擇下載,放到/home/isi/cuda目錄下面,我下載的是NVIDIA-Linux-x86_64-375.39.run

2、屏蔽開源驅(qū)動 nouveau(原文章說“可以不編輯blacklist.conf,NVIDIA.run驅(qū)動程序可以自己停止其他開源驅(qū)動,但是需要重啟”)
sudo gedit /etc/modprobe.d/blacklist.conf
在文件末尾添加如下內(nèi)容
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
(這里有一行空格)
3、刪除舊NVIDIA驅(qū)動
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ū)動無法正常安裝
輸入用戶名密碼登錄,然后鍵入命令后執(zhí)行(數(shù)字不能用小鍵盤輸入)
sudo service lightdm stop
成功關(guān)閉服務(wù)后將出現(xiàn):lightdm stop/waiting
7、進(jìn)入驅(qū)動所在的文件夾,這里就是/home/isi/cuda,執(zhí)行
sudo sh NVIDIA*.run
后面就一路Accept就可以~報錯The distribution-provided pre-install script failed!不必理會,繼續(xù)安裝。最重要的一步,安裝程序問你是否使用nv的xconfig文件,這里一點(diǎn)要選yes,否則在啟動x-window時不會使用nv驅(qū)動。
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列表,則說明驅(qū)動安裝成功了
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í)行后會有一系列提示讓你確認(rèn),但是注意,有個讓你選擇是否安裝nvidia361驅(qū)動時,一定要選擇否,因為前面我們已經(jīng)安裝了更加新的nvidia367,所以這里不要選擇安裝。其余的都直接默認(rèn)或者選擇是即可。
3、安裝成功后會出現(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)境變量和動態(tài)鏈接庫,在命令行輸入:
sudo gedit /etc/profile
在打開的文件末尾加入:
exportPATH=/usr/local/cuda/bin:$PATH
6、保存之后,創(chuàng)建鏈接文件:
sudo gedit /etc/ld.so.conf.d/cuda.conf
在打開的文件中添加如下語句:
/usr/local/cuda/lib64
然后執(zhí)行下列代碼使鏈接立即生效
sudo ldconfig
新聞標(biāo)題:Ubuntu 16.04+CUDA8.0+Theano深度學(xué)習(xí)環(huán)境搭建
瀏覽地址:http://www.dlmjj.cn/article/dhhjcoc.html


咨詢
建站咨詢
