您的位置:首页 > Web前端

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

2014-11-11 18:04 309 查看


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,大功告成!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: