一、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安装完毕。