Caffe+Ubuntu 13.10+ CUDA6.5 安装指南

1.安装CUDA6.5

下载地址:https://developer.nvidia.com/cuda-downloads  找到Ubuntu对应型号的.run版本下载

下载完之后切换到控制台(Ctrl+Alt+F1),登陆之后把桌面环境给关了:

sudo service lightdm stop

然后执行刚才下载的文件

sudo sh 文件名.run

安装完成之后重新打开桌面环境就好了

sudo service lightdm start

2.安装Nvidia驱动340

下载地址:http://www.nvidia.cn/  找到340对应型号的.run版本下载

卸载之前安装的所有Nvidia驱动:

sudo apt-get purge nvidia*

下载完之后切换到控制台(Ctrl+Alt+F1),登陆之后把桌面环境给关了:

sudo service lightdm stop

然后执行刚才下载的文件

sudo sh 文件名.run

安装完成之后重新打开桌面环境就好了

sudo service lightdm start

3.安装ATLAS

sudo apt-get install libatlas-base-dev

4. 安装OpenCV

到Github上下载已经写好的安装脚本:https://github.com/jayrambhia/Install-OpenCV

下载该脚本,进入Ubuntu/2.4 目录, 给所有shell脚本加上可执行权限

chmod +x *.sh
然后安装最新版本 (当前为2.4.9)
sudo ./opencv2_4_9.sh
脚本会自动安装依赖项,下载安装包,编译并安装OpenCV。整个过程大概半小时左右。

5.安装其他依赖项

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

其他版本用户参考官方说明:http://caffe.berkeleyvision.org/installation.html

6. 安装Caffe所需要的Python环境

首先安装pip和python-dev (系统默认有python环境的, 不过我们需要的使python-dev)
sudo apt-get install python-dev python-pip

然后执行如下命令安装编译caffe python wrapper 所需要的额外包
sudo pip install -r /path/to/caffe/python/requirements.txt

在执行上述命令时, 会报错导致不能完全安装所有需要的包。 可以按照官方建议安装anaconda包。 在anaconda官网下载.sh文件,执行,最后添加bin目录到环境变量即可。

7. 编译Caffe

进入caffe根目录, 首先复制一份Makefile.config
cp Makefile.config.example Makefile.config

编译:
make all
make test
make runtest

在make时可能会出现error:
src/caffe/util/math_functions.cu(140): error: calling a host function("std::signbit ") from a globalfunction("caffe::sgnbit_kernel ") is not allowed
解决方案:
修改  ./include/caffe/util/math_functions.hpp 224行
删除(注释):using std::signbit;
修改:DEFINE_CAFFE_CPU_UNARY_FUNC(sgnbit, y[i] = signbit(x[i]));
为:DEFINE_CAFFE_CPU_UNARY_FUNC(sgnbit, y[i] = std::signbit(x[i]));

编译Caffe Python
make pycaffe



然后在~/.bashrc中将添加caffe/python的路径:
export PYTHONPATH=/path/to/caffe/python
最后sudo ldconfig确认。


OK,大功告成!





  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值