Ubuntu14.04_x64_Caffe_GPU环境配置

为了让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
完成...

巴特西

Ubuntu14.04_x64_Caffe_GPU环境配置

最新文章

热门文章