Ubuntu14.04+CUDA8.0+TitanX+caffe配置教程

因之前已安装好tensorflow的GPU版,现因需要再安装caffe。

系统信息:
Ubuntu 14.04
TiTanX

已安装的有:

  • Cuda8.0 ( Linux x86_64 Unumtu 14.04 )
  • cudnn 5.0
  • python2.7
  • tensorflow1.0
  • tflearn
  • gcc 4.8.4

上面安装cuda8.0和cudnn5.0具体可参考:
http://blog.csdn.net/frank_zrh/article/details/52298909
http://blog.csdn.net/yan_song_/article/details/53154611

caffe和opencv安装

opencv3.0安装

这里推荐网友写的一个脚本(百度云下载http://pan.baidu.com/s/1qX1uFHa ,密码:wysa)
在Isstall-OpenCV-master的根目录下:

cd ./Ubuntu
sudo sh dependencies.sh
cd ./3.0
sudo sh opencv3_0_0.sh

**下载后,这里会出现报错,github大神给出了解决方案
https://github.com/opencv/opencv/pull/6510/commits/10896129b39655e19e4e7c529153cb5c2191a1db
因为现在opencv3.0有个地方还未修改,遇到CUDA8.0会

cd opencv-3.0.0/modules/cudalegacy/src
sudo vim graphcuts.cpp

**在代码line45中找到

#if !defined (HAVE_CUDA) || defined (CUDA_DISABLER)

改成:

#if !defined (HAVE_CUDA) || defined (CUDA_DISABLER) || (CUDART_VERSION >= 8000)

这样再编译一下就好了,再运行

sudo sh opencv3_0_0.sh 

中间出现两个问题:
1.nvcc warning : The ‘compute_20’, ‘sm_20’, and ‘sm_21’ architectures are deprecated……
这个警告不影响后面编译结果。
2./usr/local/cuda-8.0/targets/x86_64-Linux/lib/libcudnn.so.5 不是符号连接‘’不是符号连接。
解决办法:
在终端输入:
sudo ldconfig -v
找到这一行错误:libcudnn.so.5 -> libcudnn.so.5.1.10
是这个链接错误,然后在终端输入:
sudo ln -sf /usr/local/cuda-8.0/targets/x86_64-linux/lib/libcudnn.so.5.1.10 /usr/local/cuda-8.0/targets/x86_64-linux/lib/libcudnn.so.5
问题就解决了。
或者直接去目录下找:
这里写图片描述
安装成功可查看opencv版本信息:
这里写图片描述

caffe安装

*1)安装依赖项和各种库:

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler libatlas-base-dev libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install --no-install-recommends libboost-all-dev

*安装pycaffe所需要的依赖项

sudo apt-get install -y python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags cython ipython

2) BLAS 安装

sudo apt-get install libatlas-base-dev

3)去caffe的github下载caffe源码包

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

注意:若没有安装Git,需要先安装Git:

sudo apt-get install git

4)进入caffe下的python目录,安装requirements.txt文件里需要的依赖包,防止缺少依赖包:

cd caffe/python
for req in $(cat requirements.txt); do pip install $req; done

5)将Makefile.config.example的内容复制到Makefile.config:

sudo cp Makefile.config.example Makefile.config
sudo vim 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

因为我这个电脑没有安装matlab,所以不用修改MATLAB_DIR := /usr/local/MATLAB/R2014a。

5)编译caffe

sudo make all -j4
sudo make test -j4
sudo make runtest -j4
sudo make pycaffe

完成编译。
不知道自己电脑的CPU核数,可使用lscpu查看:
这里写图片描述
我的即为4,所以-j4。

测试caffe

cd ~/caffe
 sudo sh data/mnist/get_mnist.sh
 sudo sh examples/mnist/create_mnist.sh
 sudo sh examples/mnist/train_lenet.sh

caffe 例程结果

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值