为了让Caffe支持GPU模式,需要安装GPU和CUDA Toolkit,依据NVIDIA官方教程以ubuntu14.04_X64安装cuda8.0,配置Gpu为例如下所示:
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
完成...
原文:http://www.cnblogs.com/xzyzg/p/6510846.html