首页 > 系统服务 > 详细

Ubuntu14.04_x64_Caffe_GPU环境配置

时间:2017-03-06 17:24:09      阅读:240      评论:0      收藏:0      [点我收藏+]

为了让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环境配置

原文:http://www.cnblogs.com/xzyzg/p/6510846.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!