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
  完成...

posted @ 2017-03-06 17:08  心中月正古  阅读(565)  评论(0编辑  收藏  举报