一、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安裝完畢。
原創文章,作者:UOCL,如若轉載,請註明出處:https://www.506064.com/zh-hant/n/131623.html