Ubuntu14.04_x64_Caffe_GPU環境配置

爲了讓Caffe支持GPU模式,須要安裝GPU和CUDA Toolkit,依據NVIDIA官方教程以ubuntu14.04_X64安裝cuda8.0,配置Gpu爲例以下所示:linux

1.檢查安裝環境,是否具有N卡
 $ spci | grep -i nvidia
2.檢查系統版本
 $ uname -m && cat /etc/*release
   例如:
   x86_64
   Red Hat Enterprise Linux Workstation release 6.0 (Santiago)
3.驗證系統有正確的Kernel Headers
   $ uname -r
4.安裝Kernel Headers,以及環境包
   $ sudo apt-get install linux-headers-$(uname -r)
5.進入管網下載驅動
    網址:  http://developer.nvidia.com/cuda-downloads
    選擇合適的版本,例如:linux-x86_64-ubuntu-14.04-deb(local)
6.安裝命令:
    sudo dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb
    sudo apt-get update
    sudo apt-get install cuda
 沒報錯就是安裝ok
7.配置環境
  個人系統是64位的,配置環境
       export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
7.安裝sample
    $ cuda-install-samples-8.0.sh ./
9.切換到GPU模式
  打開Makefile.config
  # CPU_ONLY :=1 取加上註釋#(以前是用的cpu),保存退出
  $ make clean
  $ make -j16
  完成...ubuntu

相關文章
相關標籤/搜索