您的位置:

caffe安装详解

一、caffe安装GPU版本

在安装GPU版本之前,需要确认一下系统是否支持CUDA。可以通过以下命令查看:

nvcc -V

如果输出显示“command not found”,则需要安装CUDA。具体安装方法请参考官网教程。

1、从caffe官网(http://caffe.berkeleyvision.org/install_apt.html)下载相应版本的caffe安装文件

sudo apt-get update
sudo apt-get upgrade

2、下载相关依赖包

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install libatlas-base-dev libopenblas-dev libgflags-dev libgoogle-glog-dev liblmdb-dev

3、解压caffe安装文件并编译

tar -xvf caffe>version<.tar.gz
cd caffe>version<
cp Makefile.config.example Makefile.config
sudo vim Makefile.config

修改Makefile.config文件中相应的配置(比如开启GPU支持等),保存并退出。

make all -j8
make test -j8
make runtest -j8
sudo make pycaffe -j8

二、conda安装caffe

1、在anaconda环境下新建虚拟环境。如果你没有安装anaconda,可以到官网下载并安装

conda create --name py27 python=2.7
conda activate py27

2、添加anaconda的安装目录到环境变量PATH中

echo export PATH=~/anaconda3/bin:$PATH >> ~/.bashrc
source ~/.bashrc

3、安装caffe。可以通过所需版本github地址来获取caffe

conda install caffe

三、caffe安装笔记本电脑

1、下载需要的依赖包

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install python-dev python-numpy python-py python-pytest python-coverage python-pip libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-dev protobuf-compiler gfortran libjpeg62 libfreeimage-dev libatlas-base-dev git python-dev python-setuptools python-numpy python-scipy libprotobuf-dev libleveldb-dev liblmdb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo pip install -U pip protobuf
sudo pip install --user --install-option="--prefix=" -r python/requirements.txt

2、下载caffe

git clone https://github.com/BVLC/caffe.git

3、添加caffe路径

echo "export PYTHONPATH=$(pwd)/python:$PYTHONPATH" >> ~/.bashrc
source ~/.bashrc

4、编译caffe

cd caffe
mkdir build
cd build

打开CPU版本

cmake -DCPU_ONLY=ON ..

或者打开GPU版本

cmake -DUSE_CUDNN=OFF -DUSE_NCCL=OFF ..

编译

make all -j8
make install
make runtest

四、离线安装caffe

1、到caffe官网下载相应版本的离线安装包

2、解压压缩包并进入解压后的目录

tar -zxvf caffe>version<.tar.gz
cd caffe>version<

3、修改Makefile.config文件,添加相应选项

4、安装必要依赖项

sudo apt-get update
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install libatlas-base-dev libopenblas-dev libgflags-dev libgoogle-glog-dev liblmdb-dev

5、编译caffe

make all
make test
make runtest

至此,caffe安装完毕。