Ubuntu15.10 + Cuda7.5 配置Caffe

最新推荐文章于 2020-05-08 13:34:02 发布

qiexingqieying

最新推荐文章于 2020-05-08 13:34:02 发布

阅读量1.1k

点赞数 1

分类专栏： deep learning 文章标签： deep learning ubuntu cuda caffe 深度学习

本文链接：https://blog.csdn.net/qiexingqieying/article/details/51800458

版权

deep learning 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

最近帮导师配置了新买的笔记本上的caffe。

之前的环境都是在师兄们已经配置好的服务器上的，这次是从无到有，自己完完全全配置了一次。在这里写下总结。

主要参考的资料有：

Ubuntu 16.04 or 15.10 Installation Guide

https://github.com/BVLC/caffe/wiki/Ubuntu-16.04-or-15.10-Installation-Guide

wangruohui/Caffe Ubuntu 15.10.md

https://gist.github.com/wangruohui/679b05fcd1466bb0937f

Caffe + Ubuntu 15.04 + CUDA 7.5 新手安装配置指南

http://ouxinyu.github.io/Blogs/20151108001.html

可能因为笔记本的硬件较新，最初装Ubuntu 14.04的时候，无线和有线的驱动都有问题。所以选用更新的Ubuntu 15.10。

1. 安装Ubuntu操作系统

下载好15.10的.iso文件，用UltraISO工具制作成U盘引导盘。

安装好后，更新一下源：

sudo apt-get update

sudo apt-get upgrade

2. 安装Chrome

为了便于翻墙查找资料，选用Chrome浏览器。

2.1 先从官网下载.deb文件

2.2 sudo dpkg -i xxx.deb安装

显示libappindicator1 error

sudo apt-get -f install libappindicator1 libindicator7 进行安装

3. 安装caffe依赖包

sudo apt-get install -y build-essential cmake git pkg-config

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

sudo apt-get install -y libatlas-base-dev 

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

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

# (Python general)
sudo apt-get install -y python-pip

# (OpenCV 2.4)
sudo apt-get install -y libopencv-dev

sudo apt-get install python-opencv

一定要保证这些包都安装好。

4. 安装Nvidia驱动和Cuda

4.1 Nvidia驱动

sudo apt-get install nvidia-352-updates nvidia-modprobe

在ternimal运行命令 nvidia-smi，显示显卡的情况表示可以了。

4.2 Cuda安装

在https://developer.nvidia.com/cuda-downloads 下载cuda7.5后

sudo dpkg -i cuda-repo-ubuntu1504_7.5-18_amd64.deb

sudo apt-get update

sudo apt-get install cuda

之后在~/.bashrc文件中加入cuda路径。export LD_LIBRARY_PATH=$LD_LIBRARY_PATH: /usr/local/cuda/lib64

不然会报找不到cuda的错误

error while loading shared libraries: libcudart.so.7.5: cannot open shared object file: No such file or directory

然后 source .bashrc，使.bashrc文件立刻生效

5. 更改caffe的config文件，编译caffe

下载caffe，进入caffe目录

cp Makefile.config.example Makefile.config

在Makefile.config进行更改：

PYTHON_INCLUDE := /usr/include/python2.7 /usr/lib/python2.7/dist-packages/numpy/core/include

WITH_PYTHON_LAYER := 1

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

进入caffe目录，安装python 依赖包

cd python

for req in $(cat requirements.txt); do pip install $req; done

处理hdf5命名问题：

cd /usr/lib/x86_64-linux-gnu

sudo ln -s libhdf5_serial.so.10.0.1 libhdf5.so

sudo ln -s libhdf5_serial_hl.so.10.0.1 libhdf5_hl.so

注意！这里的10.0.1序号，最好进入x86_64-linux-gnu目录看看，填写合适的序号。

使cuda7.5支持15.10的gcc

ubuntu15.10的默认gcc是5.2，而cuda7.5只支持gcc4.9以下的，会报这类错误

In file included from /usr/local/cuda/include/cuda_runtime.h:76:0,
                 from <command-line>:0:
/usr/local/cuda/include/host_config.h:115:2: error: #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!

做法：找到 /usr/local/cuda/include/host_config.h，注释掉( // ) 115行。

最后进行编译：

cd ..

(now you are in caffe-master directory)

make all  -j8

make test -j8

make runtest

make pycaffe      -should be finished already, so you can omit this one

-j表示指定编译的cpu核数，cpu多的，可以写大点的数。

最后将python caffe加入到路径中。

同样的，打开~/.bashrc文件。