爲了讓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