linux+caffe+cuda6.5+GPU

本文是接的我的上一篇博客的後續:

blog.csdn.net/dachao_xu/article/details/46544043    linux+caffe+cpu配置


之前是因爲沒有買顯卡,所以無奈配置了linux+caffe+cpu。雖然能跑,但是顯卡驅動和cuda都沒有裝,基本上沒裝什麼東西。本文章也主要寫這兩部分的安裝。


1.顯卡驅動的安裝

將nouveau添加到黑名單,防止它啓動

$ cd /etc/modprobe.d

$ sudo gedit nvidia-graphics-drivers.conf

寫入:blacklist nouveau

檢查:$ cat nvidia-graphics-drivers.conf


 對於:/etc/default/grub,添加到末尾。
$ sudo gedit /etc/default/grub
末尾寫入:rdblacklist=nouveau nouveau.modeset=0

檢查:$ cat /etc/default/grub


2.進入Ubuntu, 按 ctrl+alt+F1 進入tty1, 登錄tty1後

輸入如下命令(注意以下操作均在這個黑色界面裏操作,在輸入password的時候請用字母上方的數字鍵盤!!)

$ sudo service lightdm stop

安裝NVIDIA ,http://www.geforce.cn/drivers 找到對應NVIDIA的版本(我的是linux340.76.run),進入到NVIDIA的目錄下:

$ cd /home/xc                注:(你的用戶名)

$ sudo sh NVIDIA-Linux-x86_64-340.76.run     注:(你要cd它所在的文件夾)

一直確定安裝完。

重新啓用桌面的命令爲:$ sudo service lightdm start

3.驗證驅動安裝是否成功

$ cat /proc/driver/nvidia/version     注:cat後面有空格!

4.CUDA6.5安裝(https://developer.nvidia.com/cuda-downloads

$ sudo chmod +x ./ cuda_6.5.14_linux_64.run

將CUDA6.5解壓成三個文件($sudo sh cuda*.run --noexec --target .)(.前面有空格)

別爲:CUDA安裝包: cuda-linux64-rel-6.5.14-18749181.run

    NVIDIA驅動: NVIDIA-Linux-x86_64-340.29.run   (這裏就不安裝NVIDIA驅動)

    SAMPLE包: cuda-samples-linux-6.5.14-18745345.run

5.開始安裝CUDA

$ sudo ./cuda-linux64-rel-6.5.14-18749181.run

安裝完成後需要在/etc/profile中

添加環境變量, $ PATH=/usr/local/cuda-6.5/bin:$PATH$ export PATH

保存後, 執行下列命令, 使環境變量立即生效$ source /etc/profile

添加lib路徑,在/etc/ld.so.conf.d/新建文件cuda.conf,並編輯

cd /etc/ld.so.conf.d

$ sudo touch cuda.conf

$ sudo gedit cuda.conf

/usr/local/cuda-6.5/lib64   注:即在空白文檔里加這一行,保存。

執行下列命令使之立刻生效  

$ sudo ldconfig

6.安裝CUDA SAMPLE

安裝下列依賴包

$ sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

然後用下述命令安裝sample文件

$ sudo ./cuda-samples-linux-6.5.14-18745345.run

完成後編譯Sample文件,這邊編譯的時間有點久,請耐心等待!

編譯完成後

$ cd /usr/local/cuda-6.5/samples

$ sudo make

編譯完成後, 進入 samples/bin/x86_64/linux/release

$ sudo ./deviceQuery

如果出現下列顯卡信息,則驅動及cuda安裝成功:

./deviceQuery Starting...

 CUDA Device Query (Runtime API) version (CUDART static linking)

Detected 1 CUDA Capable device(s)

Device 0: "GeForce GTX 750 Ti"
 CUDA Driver Version / Runtime Version          6.5 / 6.5
 CUDA Capability Major/Minor version number:    5.0
 Total amount of global memory:                 2048 MBytes (2147155968 bytes)
 ( 5) Multiprocessors, (128) CUDA Cores/MP:     640 CUDA Cores
 GPU Clock rate:                                1084 MHz (1.08 GHz)
 Memory Clock rate:                             2700 Mhz
 Memory Bus Width:                              128-bit
 L2 Cache Size:                                 2097152 bytes
 Maximum Texture Dimension Size (x,y,z)         1D=(65536), 2D=(6553665536), 3D=(4096, 4096, 4096)
 Maximum Layered 1D Texture Size, (num) layers  1D=(16384), 2048 layers
 Maximum Layered 2D Texture Size, (num) layers  2D=(1638416384), 2048 layers
 Total amount of constant memory:               65536 bytes
 Total amount of shared memory per block:       49152 bytes
 Total number of registers available per block: 65536
 Warp size:                                     32
 Maximum number of threads per multiprocessor:  2048
 Maximum number of threads per block:           1024
 Max dimension size of a thread block (x,y,z): (1024, 1024, 64)
 Max dimension size of a grid size    (x,y,z): (21474836476553565535)
 Maximum memory pitch:                          2147483647 bytes
 Texture alignment:                             512 bytes
 Concurrent copy and kernel execution:          Yes with 1 copy engine(s)
 Run time limit on kernels:                     Yes
 Integrated GPU sharing Host Memory:            No
 Support host page-locked memory mapping:       Yes
 Alignment requirement for Surfaces:            Yes
 Device has ECC support:                        Disabled
 Device supports Unified Addressing (UVA):      Yes
 Device PCI Bus ID / PCI location ID:           1 / 0
 Compute Mode:
    < Default (multiple host threads can use ::cudaSetDevice() with device simultaneously) >

deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 6.5, CUDA Runtime Version = 6.5, NumDevs = 1, Device0 = GeForceGTX 750 Ti
Result = PASS


7.caffe的重新編譯

由於之前在只有cpu的時候已經編譯過來caffe,我以爲會ok,make all沒什麼問題,但在make test的時候出錯了。解決辦法很簡單,就是先make clean下,再去

$ cd caffe-master

$ sudo make all

$ sudo make test

$ sudo make runtest

再用相同的方法去測試minst,運行成功那麼整個linux+caffe+cuda6.5+GPU+opencv+matlab+python就已經配置好了。

撒花~~~~~~~~~~~




感謝langb2014對我的幫助,感謝!



發佈了21 篇原創文章 · 獲贊 3 · 訪問量 7萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章