スチールウールの活動記録

初心者に向けて(私もだけど)電子工作やプログラミングや関連することをいろいろ。

caffeのインストール

Deep Learningを使えるCaffeをインストールする手順。

CUDAをインストール

sudo apt-get install nvidia-cuda-toolkit

BLASをインストール

sudo apt-get install libatlas-base-dev

その他をインストール

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

FORTRANのインストール

sudo apt-get install gfortran

Caffeをgitからクローン

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

Caffeのディレクトリに移動し、Makefile.config.exampleMakefile.configに名前を変更

cp Makefile.config.example Makefile.config

Makefile.configの中を書き換える

# CPU_ONLY := 1 → CPU_ONLY := 1
# CUSTOM_CXX := g++ → CUSTOM_CXX := g++-
CUDA_DIR := /usr/local/cuda → CUDA_DIR := /usr/lib/nvidia-cuda-toolkit

makeする

make all
make test
make runtest

pip, numpy, skimageなどを入れる

sudo apt-get install python-dev python-pip python-numpy python-skimage

pipを実行

sudo pip install -r ~/caffe/python/requirements.txt

makeする

make pycaffe

パスを通す

vi ~/.bashrc  
export PYTHONPATH=~/caffe/python/:$PYTHONPATH
source ~/.bashrc

最後に

python
import caffe

が成功すれば完了(なはず)