• UID80
  • 登录2017-11-20
  • 粉丝63
  • 关注21
  • 发帖1124
  • 主页
  • 金币6473枚
社区居民
忠实会员
原创写手
潇潇宇 发布于2017-07-12 16:16
1/282

双显卡UBuntu16.04安装Caffe

楼层直达

双显卡UBuntu16.04安装Caffe

 

双显卡UBuntu16.04安装Caffe

2017年2月2日记

Ubuntu16.04 + CUDA8.0 + CUDNN +opencv3.1 + Caffe

作者已经装过N遍caffe了,苦逼的原因是因为双显卡的电脑,集成显卡+独立显卡(ntel核显以及Nvidia GEFORCE GTX860M显卡),在Ubuntu14.04下安装CUDA会出现比如循环登录、启动后黑屏等问题。比如如下问题(所以为了避免入坑,赶紧装Ubuntu16.04):

1.启动后电脑黑屏,不能进行任何操作(可能还有救)。

在grub界面选择要登录的系统按“e”,进入编辑界面,在倒数第二行末尾加入acpi_osi=Linux nomodeset,具体操作可以上网查。

2.启动后电脑黑屏但是可以通过CTRL+ALT+F1(我的电脑是CTRL+ALT+F4)进入命令行模式(可能还有救)。

删除所有nvidia,sudo apt-get remove  nvidia*,然后重启。

3.进入登录界面,然后不断地让你登录,一直登录不进去,进入命令行也一直闪烁(可能没救了)。

但是这个问题在Ubuntu16.04就得到了解决,所以,如果你的电脑是双显卡的话,just do it!

忽略上面的命令啊,那个只是Ubuntu14.04的抢救措施。装好了Ubuntu16.04,开始吧:

1.首先安装CUDA

a.首先安装一些依赖

sudo apt-get update

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

sudo apt-get install --no-install-recommends libboost-all-dev

sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

sudo apt-get install git cmake build-essential

b.安装好之后需要禁用nouveau,创建/etc/modprobe.d/blacklist-nouveau.conf,写入:

blacklist nouveau option nouveau modeset=0

然后执行sudo update-initramfs -u

c.配置一些基本的环境变量

vim ~/.bashrc

加入:

export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH

到这里之后有些教程会让我们上NVIDIA官网上看我们的GPU型号应该用什么版本的驱动,其实不需要;另外有些教程会让我们把gcc和g++版本降低,比如降低为gcc-4.9和g++-4.9,其实不需要,Ubuntu16.04的默认gcc和g++版本就为5.4版本,而CUDA8.0目前也支持此版本。

d.然后我们需要下载CUDA(https://developer.nvidia.com/cuda-downloads),这个网址

image

e.安装cuda有点讲究

下载完成后重启电脑,在出现登录界面时,登录tty1,按Ctrl + Alt + F1,在文本模式中登录.

关闭桌面服务

sudo service lightdm stop。

这步对于安装Nvidia驱动至关重要

进入下载cuda文件的目录执行:

sudo sh cuda-8.0.44_linux.run --no-opengl-libs

如果不加–no-opengl-libs应该是会出现循环登录问题

安装过程中完全按照它的指引就可以了,先输入accept,然后一连串的yes就可以。

Accept EULA conditions

Say YES to installing the NVIDIA driver

Say YES to installing CUDA Toolkit + Driver

Say YES to installing CUDA Samples等等。

然后重启,这个时候没有循环登录问题,正常进入桌面。

f.修改环境配置

vim ~/.bashrc

加上:export PATH=/usr/local/cuda-8.0/bin:$PATH

     export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

刷新配置执行

source ~/.bashrc

g.测试CUDA的samples:

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

会看到你的GPU信息,如果看到,说明CUDA安装成功。

2.配置cuDNN

cuDNN是GPU加速计算深层神经网络的库。

(a)首先去官网(https://developer.nvidia.com/rdp/cudnn-download)下载cuDNN,需要注册一个账号才能下载

image

(b)下载cuDNN5.1之后进行解压,cd进入cuDNN5.1解压之后的cuda/include目录,在命令行进行如下操作:

sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件

(c)再将cd进入lib64目录下的动态文件进行复制和链接:

sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件
sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5 #生成软衔接
sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接

3.安装opencv3.1

博主这个时候就有点SB了,OpenCV可以apt-get,我当时却自己编译的,不过还好,只有一个坑。

从官网(http://opencv.org/downloads.html)下载opencv,并将其解压到你要安装的位置。

在你的OpenCV目录下

mkdir build # 创建编译好的文件目录

cd build

配置:

cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..

编译:

make -j8 #-j8表示并行计算,根据自己电脑的配置进行设置,配置比较低的电脑可以将数字改小或不使用,直接输make。

以上只是将opencv编译成功,还没将opencv安装,需要运行下面指令进行安装:

sudo make install
  • 可能出现的错误,编译opencv遇到的坑:

image

类似此类错误,应该在进度92%处出现。

modules/cudalegacy/src/graphcuts.cpp:120:54: error: 
‘NppiGraphcutState’ has not been declared
typedef NppStatus (*init_func_t)(NppiSize oSize, 
NppiGraphcutState** ppState, Npp8u* pDeviceMem);

这是因为opecv3.1与cuda8.0不兼容导致的。解决办法:

修改 ~/opencv/modules/cudalegacy/src/graphcuts.cpp文件内容,如图:

image

4.安装Caffe

不建议安装Anaconda,我开始安装Anaconda,但是出现问题,后来使用原生态的Python就OK了。

遇到了什么问题呢?下图:

image

caffe的GitHub的issue也有问题的解决方法:

https://github.com/BVLC/caffe/issues/985

我按照GitHub所说,执行

rm ~/anaconda/lib/libm.*

后无卵用,我就用原生态的python接口了,很方便配置。所以忽略Anaconda,直接安装Caffe。

下面安装Caffe:

1)将终端cd到要安装caffe的位置。

2)从github上获取caffe:

git clone https://github.com/BVLC/caffe.git

3)因为make指令只能make Makefile.config文件,而Makefile.config.example是caffe给出的makefile例子,因此,首先将Makefile.config.example的内容复制到Makefile.config:

sudo cp Makefile.config.example Makefile.config

4)打开并修改配置文件:

sudo gedit Makefile.config #打开Makefile.config文件

修改文件:

a.若使用cudnn,则:

将
#USE_CUDNN := 1
修改成: 
USE_CUDNN := 1

b.因为opencv版本是3的,则

将
#OPENCV_VERSION := 3 
修改为: 
OPENCV_VERSION := 3

c.若要使用Python来编写layer,则

将
#WITH_PYTHON_LAYER := 1 
修改为 
WITH_PYTHON_LAYER := 1

d. 重要的一项 :

将# Whatever else you find you need goes here.下面的

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 
修改为: 
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial       

这是因为ubuntu16.04的文件包含位置发生了变化,尤其是需要用到的hdf5的位置,所以需要更改这一路径.

5)修改makefile文件,打开makefile文件,做如下修改:

将:
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
替换为:
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

在Makefile文件的第173行,把 hdf5_hl 和hdf5修改为hdf5_serial_hl 和 hdf5_serial,也就是把下面第一行代码改为第二行代码。

将:
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
改为:
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

6)编辑/usr/local/cuda/include/host_config.h

将其中的第115行注释掉:

将
#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!
改为
//#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!

7)编译

make all -j8 #-j根据自己电脑配置决定

sudo make runtest -j8 # 跑test

image

sudo make pycaffe -j8 #编译pycaffe

然后,配置环境变量,将caffe/python加到环境变量里去,然后输入Python:

import caffe

image

开开心心去玩耍吧!

Reference :

1.http://blog.csdn.net/xuzhongxiong/article/details/52717285

2.http://blog.csdn.net/t5131828/article/details/53258925


0人打赏
  • UID80
  • 登录2017-11-20
  • 粉丝63
  • 关注21
  • 发帖1124
  • 主页
  • 金币6473枚
社区居民
忠实会员
原创写手
潇潇宇 发布于2017-07-15 09:55
沙发F
您需要登录后才可以回帖
发表回复
极贡献
技术问答
专题荟萃
程序人生
视觉设计
Android开发
iOS开发
编程语言
前端开发
后端开发
服务器架构
软件测试
运维方案
创业路上



最热文章墙

  • 79483/378   【精品推荐】200多种Android动画效果的强悍框架,太全了,不看这个,再有动画的问题,不理你了^@^

  • 45334/191   情人节福利,程序员表白的正确姿势:改几行代码就变成自己的表白了

  • 45016/0   Python爬虫:常用浏览器的useragent

  • 41525/260   【精品推荐】Android版产品级的音乐播放器源码,功能太强大了,最好的产品原型有木有?

  • 38705/145   省时省力的Android组件群来了,非常棒的原型参考

  • 30086/142   2016抢红包软件及源码

  • 29311/2   超全!整理常用的iOS第三方资源

  • 29302/71   原创表白APP,以程序员的姿势备战新年后的7夕,持续完善中!

  • 24385/160   Android版类似UC浏览器:非常赞,产品级的源码

  • 22882/30   麻省理工的一帮疯子,真的实现了随意操控万物!(绝对黑科技)

  • 22578/25   Android工程师面试题大全

  • 22502/27   2016程序员跳槽全攻略

  • 21959/9   GitHub上排名前50的iOS项目:总有一款你用得着

  • 20927/20   码魂:程序员的牛B漫画

  • 19088/85   Android小而全的博客源码:非常适合全面掌握开发技巧

  • 19064/73   【持续更新中】Android福利贴(二):资料源码大放送

  • 19030/10   2016年最全的Android面试考题+答案 精编版

  • 18936/42   一个绚丽的loading动效分析与实现!

  • 18771/3   吐槽那些程序员的搞笑牛逼注释

  • 17658/45   惊艳的App引导页:背景图片切换加各个页面动画效果

  • 17636/1   iOS 动画总结

  • 17526/82   仿京东商城客户端Android最新版,不错的原型和学习资料

  • 17491/104   Android带弹幕的视频播放器源码,来自大名鼎鼎的Bilibili弹幕网站

  • 17165/23   个人收集的Android 各类功能源代码

  • 16703/5   新一代Android渠道打包工具:1000个渠道包只需要5秒

  • 16627/21   Android福利第三波【Android电子书】

  • 16449/53   基于瀑布流的美女图片浏览App,有注释的源代码

  • 16440/81   【精品推荐】类似360安全卫士安Android源码:非常赞的产品原型

  • 16411/10   女程序员的梦,众网友的神回复

  • 16387/0   iOS中文版资源库,非常全

  • 16350/17   用JavaScript 来开发iOS和Android 原生应用:React Native开源框架中文版来啦

  • 16097/11   年会上现场review代码是怎么样的体验!

  • 16050/23   珍藏多年的素材,灵感搜寻网站

  • 15540/18   65条最常用正则表达式,你要的都在这里了

  • 14859/15   基于Android支付宝支付设计和开发方案

  • 14271/11   有木有这样一张酷图帮你集齐所有git命令超实用

  • 14198/17   什么是真正的黑客:收获12200+Stars,人气远超微软开源VS

  • 13988/46   在线音乐播放器完整版(商用级的源码):非常赞,可听免费高品质专辑

  • 13728/0   GitHub iOS 库和框架Top100 

  • 13691/61   【技巧一】搭配Android Studio,如何实现App远程真机debug?

  • 13573/7   用程序员的姿势抢过年的火车票

  • 13527/7   一张图搞定iOS学习路线,非常全面

  • 13155/10   成为Java顶尖程序员 ,看这11本书就够了

  • 13115/10   微信支付终于成功了(安卓,iOS),在此分享

  • 13010/18   一张图搞定Android学习路线,非常全面

  • 12828/29   【持续更新中】Android福利贴(一):资料源码

  • 12751/3   基于Node.js的强大爬虫,能直接发布抓取的文章哦

  • 12503/4   46 个非常有用的 PHP 代码片段

  • 12024/3   即时通信第三方库

  • 11456/8   流媒体视频直播方案

  • 11390/18   八个最优秀的Android Studio插件

  • 11263/9   B站建开源工作组:APP想支持炫酷弹幕的看过来

  • 11048/9   烧了5亿美金,这家神秘的公司即将颠覆人类未来!

  • 11047/2   【精品推荐】高质量PHP代码的50个实用技巧:非常值得收藏

  • 10958/10   中国黑客的隐秘江湖:攻守对立,顶尖高手月入千万美元

  • 10232/6   开箱即用!Android四款系统架构工具

  • 10023/10   十大技巧快速提升Android应用开发性能

  • 9985/3   10款GitHub上最火爆的国产开源项目——可以媲美西半球

  • 9931/1   Android性能优化视频,文档以及工具

  • 9842/0   基于node-webkit跨平台应用案例集之(一)

  • 返回顶部