Ubuntu14.04配置caffe
前段时间接到任务需要配置caffe并在上面训练神经网络相关,对caffe与深度学习一窍不通的我有幸得到了
xizero00师兄的指导并在Ubuntu 14.04上面配置了caffe,亲测可用,以下是师兄写的博文正文:
网上充斥各种安装方法,但是都很繁琐,特别是安装显卡驱动以及依赖项,对于初学者不是很方便。我觉得采用包管理工具更方便因此写下本文。
(1)首先安装ssh,这样能够使用winscp和putty或者secure crt进行远程连接,方便在windows下载东西传到linux下
sudo apt-get install ssh
(2)安装caffe的依赖项
1)一般依赖项
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
2)BLAS依赖项
sudo apt-get install libatlas-base-dev
或者安装
sudo apt-get install libopenblas-dev
3)python依赖项
安装python及其头文件
sudo apt-get install python
sudo apt-get install python-dev
安装python的其他依赖
sudo apt-get install python-numpy
sudo apt-get install ipython
sudo apt-get install ipython-notebook
sudo apt-get install python-sklearn
sudo apt-get install python-skimage
sudo apt-get install python-protobuf
4)谷歌glog和gflags和lmdb依赖项
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
(3)然后安装cuda和显卡驱动
用迅雷下载
然后用winscp传输到linux下,运行如下命令
sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
sudo apt-get update
sudo apt-get install cuda
1)安装完成后设置环境变量
sudo gedit /etc/bash.bashrc
在文件的末尾加入如下
export PATH=/usr/local/cuda-7.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
2)安装完毕后验证显卡驱动是否安装完毕(如果检测不到或者提示找不到文件可以重启下系统,因为刚装完显卡驱动)
cat /proc/driver/nvidia/version
3)安装完显卡驱动之后可以安装cuda的sample,<dir>就是你要安装的位置
cuda-install-samples-7.5.sh <dir>
(4)安装git,并且下载代码
sudo apt-get install git
git clone
https://github.com/BVLC/caffe.git
然后进入到源码目录
cd caffe
(5)编译Caffe
cp Makefile.config.example Makefile.config
然后修改Makefile.config设置你的python的路径和matlab的路径
注意:-j8表示使用多核进行编译,8是核心个数
make all -j8
make test -j8
make runtest
如果要编译python的绑定(你要安装好python及其依赖)
make pycaffe -j8
如果要编译matlab的绑定(你要安装好matlab)
make matcaffe -j8
附注:
cudnn这里没写。也很简单。在nvidia网站下载一个cudnn的deb包安装一下就可以了
enjoy it!
written by xizero00
http://blog.csdn.net/xizero00
以上是转载内容
附原文地址:http://blog.csdn.net/xizero00/article/details/52549883