首頁 > 軟體

Ubuntu 16.04+CUDA8.0+Theano深度學習環境搭建

2020-06-16 17:04:48

詳細分享在Ubuntu 16.04下搭建CUDA8.0+Theano深度學習環境的步驟,希望對大家有所幫助。

一、安裝nvidia驅動

1、首先去官網上檢視適合你GPU的驅動(http://www.nvidia.com/Download/index.aspx?lang=en-us),根據自己買的型號選擇下載,放到/home/isi/cuda目錄下面,我下載的是NVIDIA-Linux-x86_64-375.39.run

2、遮蔽開源驅動 nouveau(原文章說“可以不編輯blacklist.conf,NVIDIA.run驅動程式可以自己停止其他開源驅動,但是需要重新啟動”)

sudo gedit /etc/modprobe.d/blacklist.conf

在檔案末尾新增如下內容

blacklist vga16fb

blacklist nouveau

blacklist rivafb

blacklist nvidiafb

blacklist rivatv

(這裡有一行空格)

3、刪除舊NVIDIA驅動

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、關閉圖形環境,否則驅動無法正常安裝

輸入使用者名稱密碼登入,然後鍵入命令後執行(數位不能用小鍵盤輸入)

sudo service lightdm stop

成功關閉服務後將出現:lightdm stop/waiting

7、進入驅動所在的資料夾,這裡就是/home/isi/cuda,執行

sudo sh NVIDIA*.run

後面就一路Accept就可以~報錯The distribution-provided pre-install script failed!不必理會,繼續安裝。最重要的一步,安裝程式問你是否使用nv的xconfig檔案,這裡一點要選yes,否則在啟動x-window時不會使用nv驅動。

8、重新啟動 X-window 服務,並重新啟動電腦

sudo service lightdm start

sudo reboot

9、檢視顯示卡是否裝好

方法1:執行下邊程式碼,假如顯示"direct rendering: Yes",則已安裝

glxinfo | grep rendering

方法2:執行下邊程式碼,出現GPU列表,則說明驅動安裝成功了

nvidia-smi

二、安裝cuda

Ubuntu 15.04 下Caffe + + CUDA 7.0 安裝設定指南  http://www.linuxidc.com/Linux/2016-11/137497.htm

Ubuntu 16.04系統下CUDA7.5設定Caffe教學 http://www.linuxidc.com/Linux/2016-07/132859.htm

Ubuntu 16.04+CUDA7.5+Caffe深度學習環境搭建  http://www.linuxidc.com/Linux/2017-10/147612.htm

1、從這裡下載cuda的安裝檔案

https://developer.nvidia.com/cuda-release-candidate-download


 

注意這裡下載的是cuda8.0的runfile(local)檔案,我放在了/home/isi/cuda資料夾下

2、進入到/home/isi/cuda資料夾,進行安裝

cd /home/isi/cuda

sudo sh cuda_8.0.27_linux.run

執行後會有一系列提示讓你確認,但是注意,有個讓你選擇是否安裝nvidia361驅動時,一定要選擇否,因為前面我們已經安裝了更加新的nvidia367,所以這裡不要選擇安裝。其餘的都直接預設或者選擇是即可。

3、安裝成功後會出現如下介面:

===========

= 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、安裝完畢後,再宣告一下環境變數,並將其寫入到 ~/.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、然後設定環境變數和動態連結庫,在命令列輸入:

sudo gedit /etc/profile

在開啟的檔案末尾加入:

exportPATH=/usr/local/cuda/bin:$PATH

6、儲存之後,建立連結檔案:

sudo gedit /etc/ld.so.conf.d/cuda.conf

在開啟的檔案中新增如下語句:

/usr/local/cuda/lib64

然後執行下列程式碼使連結立即生效

sudo ldconfig

更多詳情見請繼續閱讀下一頁的精彩內容http://www.linuxidc.com/Linux/2017-10/147614p2.htm


IT145.com E-mail:sddin#qq.com