ubuntu14.04+GPU+caffe


非常建議ubuntu14.04的系統
一、執行下面的操作,然後驗證硬件支持GPU CUDA,只要型號存在於https://developer.nvidia.com/cuda-gpus,就沒問題了
   
 $ lspci | grep -i nvidia
     (GPU買之前是專門查過的,這個地方不太重要)
二、確定你的系統是否支持
    
$ uname -m && cat /etc/*release
重點是“x86_64”這一項,保證是x86架構,64bit系統
三、確定有gcc
 
    $ gcc --version
(網上說,沒有的話就先安裝吧,這個是必須的用來編譯CUDA Toolkit,不過Ubuntu 14.04是默認有的)
四、下載cuda和nvidia驅動(建議都下載.run格式的)
   nvidia驅動:http://www.geforce.cn/drivers
   cuda下載:https://developer.nvidia.com/cuda-downloads
   cuda由於比較大而且網內訪問較慢,需要時間有點多
五、安裝前準備,在tty中顯示中文
    因爲待會的安裝可能會有很多問題,報錯是中文的話會顯示亂碼,影響找錯。
   5.1安裝fbterm
      
sudo apt-get install fbterm
   5.2編輯.fbtermrc文件
sudo vi .fbtermrc
,加入:
font-size=16
text-codings=utf8

六、安裝nvidia顯卡和cuda(建議把他們兩個放在同一個目錄裏,待會方面找)
   6.1進入tty(ctrl+alt+f1)後輸入如下指令進入中文tty
    
sudo fbterm

   6.2退出GUI,

sudo stop lightdm

   6.3將Ubuntu系統自帶的顯卡驅動禁用,
     
    sudo vim/etc/modprobe.d/nvidia-graphics-drivers.conf
   在文件輸入:
   blacklist nouveau
   保存退出。
     
  sudo vim /etc/default/grub
  在文件末尾添加:
  rdblacklist=nouveau nouveau.modeset=0
  保存退出。
   6.4其他操作:
     
sudo mv /boot/initramfs-$(uname -r).img/boot/initramfs-$(uname -r)-nouveau.img
sudo dracut/boot/initramfs-$(uname -r).img $(uname -r)
sudo update-initramfs –u

      第一條就提示沒有這個文件,後面的就忘了,但是最最後的安裝成功,這個地方就忽略了
   6.5 cd進入nvidia驅動的目錄,安裝驅動:
    
 ls -l 查看目錄裏的內容
  sudo sh ./nvidia驅動名字

      之後安裝
   6.6 安裝cuda(不要用sudo,我的用了之後沒法跑samples,即用了之後檢測不到cuda)。另外其實這個cuda裏還有一個nvidia,因爲上面已經安裝過nvidia了,安裝cuda時就不要選它了,把軟連接(問是否要建一個/usr/local/cuda的目錄)選上,把samples選上(後面檢測是否安裝上cuda):
     
sh ./cuda名字

七、後續
   7.1回到圖形界面
    
 sudo start lightdm

   7.2修改系統環境變量
      
sudo vim ~/.bashrc

       在最後加上:    
export PATH=/usr/local/cuda-6.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-6.5/lib64:$LD_LIBRARY_PATH
sudo ldconfig(如果這句話的作用還是沒有用,重啓)

  7.3檢測cuda是否安裝成功,編譯cuda-samples並執行
     
進入你剛剛選擇cuda sample安裝的目錄後make(如果提示沒有make命令,請安裝cmake。sudo apt-get install cmake)
編譯完畢,切換release目錄cd ./bin/x86_64/linux/release
運行實例 ./deviceQuery


caffe安裝:
   1.進入這個網址按照它的指令,安裝依賴,裏面要求的cuda上面已經安裝:
     http://caffe.berkeleyvision.org/install_apt.html
   2.下載caffe:
     https://github.com/BVLC/Caffe/
   3.安裝python模塊(不影響編譯,可以略過,但是你最終還是要用到,最好安裝)
     官網:



 http://caffe.berkeleyvision.org/installation.html

因爲官網提供方法安裝不可行,可以參考我的:http://blog.csdn.net/u010668907/article/details/51112899

 4.編譯   

 官網:http://caffe.berkeleyvision.org/installation.html

     看到那個大大的Compilation以及CMake Build沒有,因爲我們是GPU,而且也沒有安裝cudnn,所以可以直接執行它的命令,不用改動配置文件(如果是cpu,就還要把CPU_ONLY := 1的註釋放開)
   5.其他的http://caffe.berkeleyvision.org/官網有很多例子等着你去看
發佈了74 篇原創文章 · 獲贊 50 · 訪問量 32萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章