转载请注明地址
说明:本文档参考自Caffe官网的安装说明,http://caffe.berkeleyvision.org/installation.html
如果对安装过程中,需要用到的依赖不明,请自行百度其作用。Ubuntu系统的安装请自行百度。
1.安装环境
硬件:Intel E3 CPU,NVIDIA GTX960 GPU; 操作系统:Ubuntu14.04
2.依赖安装
接下来的安装全部使用linux命令行操作
(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)CUDA安装
可以使用apt-get方式,或者去官网下载linux系统下的.run包进行离线安装。由于apt-get指令拿到的软件包通常不是最新的,官网推荐使用.run包的方式进行离线安装。
对于没有GPU的同学来说,可以跳过这一步(之后在编译源码时配置CPU-ONLY)。
1)下载CUDA的.run包
地址:https://developer.nvidia.com/cuda-downloads
根据自己的OS和CPU架构选择.run包并下载
2)退出桌面环境
使用键盘Ctrl+Alt+F1,系统退出桌面版,进入命令行模式。
注意:此时有些系统会遇到黑屏问题。解决方法:切回桌面版本(Ctrl+Alt+F7),修改/etc/default/grub文件中GRUB_CMDLINE_LINUX_DEFAULT的值为nomodeset,更新grub(sudo update-grub),重启电脑(或者重启电脑,在启动时进入grub设置中,将倒数第2行的roquiet splash 后面空格添加nomodeset,按F10启动系统,然后再按ctrl+alt+F1发现可以正常进入字符界面了)。
(安装时遇到这个问题,我通过后一种方式解决)
3)退出X Windows桌面环境
sudo stop lightdm
4)安装CUDA .run包
给.run包加上可执行权限:
chmod +x cuda_7.5.18_linux.run
执行.run包:
./cuda_7.5.18_linux.run
5)返回X Windows桌面环境
sudo start lightdm Ctrl+Alt+F7
6)cuDNN 安装(可选)
如果使用 cuDNN 优化神经网络, 需要安装 cudnn-v5.0 以上版本。 该软件的安装非常简单。
tar -xzvf cudnn-7.5-linux-x64-v5.0-ga.tgz cd cudnn-7.5-linux-x64-v5.0 sudo cp lib* /usr/local/cuda/lib64/ sudo cp cudnn.h /usr/local/cuda/include/
(3)BLAS
sudo apt-get install libatlas-base-dev
(4)Python
如果使用默认Python的方式进行Caffe开发,则需要安装python开发包。
sudo apt-get install python-dev
(5)兼容性说明
操作系统 |
版本 |
兼容性说明 |
Ubuntu |
16.04 |
CUDA8是被需要的 |
14.04 |
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev |
|
12.04 |
需要手动安装依赖(如下) |
如果是Ubuntu12.04,需要手动安装兼容性依赖,如下:
wget https://google-glog.googlecode.com/files/glog-0.3.3.tar.gz tar zxvf glog-0.3.3.tar.gz cd glog-0.3.3 ./configure make && make install # gflags wget https://github.com/schuhschuh/gflags/archive/master.zip unzip master.zip cd gflags-master mkdir build && cd build export CXXFLAGS="-fPIC" && cmake .. && make VERBOSE=1 make && make install # lmdb git clone https://github.com/LMDB/lmdb cd lmdb/libraries/liblmdb make && make install
3.编译Caffe源代码并安装
Caffe源码支持make编译,或者CMake方式进行编译。编译前请确认已经安装好了gcc工具链(sudo apt-get install build-essential),如果使用CMake构建工程,请确认安装好CMake。
(1)下载Caffe源码
Git clone https://github.com/BVLC/caffe.git
或者在github官网搜索Caffe,选择需要的分支然后Zip打包下载。
(2)Make编译源码
说明1:
Configure the build by copying and modifying the example Makefile.config for your setup. The defaults should work, but uncomment the relevant lines if using Anaconda Python.
以下为命令行
cp Makefile.config.example Makefile.config # Adjust Makefile.config (for example, if using Anaconda Python, or if cuDNN is desired) make all make test make runtest
说明2:
For CPU & GPU accelerated Caffe, no changes are needed.
For cuDNN acceleration using NVIDIA’s proprietary cuDNN software, uncomment the USE_CUDNN := 1 switch in Makefile.config. cuDNN is sometimes but not always faster than Caffe’s GPU acceleration.
For CPU-only Caffe, uncomment CPU_ONLY := 1 in Makefile.config.
(3)CMake方式构建工程并编译
根据需要可以配置Makefile.config。
mkdir build cd build cmake .. make all make install make runtest
如果编译没有报Error,在runtest时一切OK,那么说明安装完成。可以开始Caffe的第一个小例子MNIST。