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,大功告成!