LoginSignup
8
7

More than 5 years have passed since last update.

pyenvを用いたcaffeの導入

Last updated at Posted at 2016-04-16

2016.3.22

機械学習をする上でよく使用するライブラリを一括導入します.
環境はUbuntu14.04 LTS(これはcaffe,chainerの推奨環境なため)

pyenv

git clone https://github.com/yyuu/pyenv.git ~/.pyenv
echo 'export PYENV_ROOT="$HOME/.pyenv"' >> ~/.bashrc
echo 'export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bashrc
echo 'eval "$(pyenv init -)"' >> ~/.bashrc
source ~/.bashrc

python

pyenv install anaconda-2.1.0
pyenv rehash
pyenv global anaconda-2.1.0
echo 'export PATH="$PYENV_ROOT/versions/anaconda-2.1.0/bin/:$PATH"' >> ~/.bashrc
source ~/.bashrc

今回はCaffe 推奨のanaconda-2.1.0を使用.またanacondaは数値計算パッケージなので,導入時からnumpyやscipy,scikit-learnが入っているので便利

次に深層学習のライブラリの中から普段使うCaffe,Chainer,TensorFlowを導入

Caffe

  • cuda
## cuda install 
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_7.0-28_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1404_7.0-28_amd64.deb
sudo apt-get update
sudo apt-get install cuda
  • 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 libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler g++-4.6
  • Caffe(本体)
## caffe
cd $HOME
git clone https://github.com/BVLC/caffe.git ~/caffe
cd caffe
cp Makefile.config.example Makefile.config
### ※
echo "CPU_ONLY := 1" >> Makefile.config
echo "CXX := /usr/bin/g++-4.6" >> Makefile.config

一通り入れ終わったので,実行テストをします

make -j4 all
make -j4 test
make -j4 runtest

caffeをpythonで用いたいのでpycaffeを導入します.

##pycaffe
pip install -r ~/caffe/python/requirements.txt
sudo apt-get install python-dev python-pip python-numpy python-skimage
sudo apt-get install --no-install-recommends libboost-all-dev
echo "export PYTHONPATH=~caffe/python/:$PYTHONPATH" >> ~/.bashrc
source ~/.bashrc
make pycaffe
import caffe

で特にエラーがなければ大丈夫です。

なお

ここで

make pycaffe
CXX/LD -o python/caffe/_caffe.so python/caffe/_caffe.cpp
touch python/caffe/proto/__init__.py
PROTOC (python) src/caffe/proto/caffe.proto

>>> import caffe
Traceback (most recent call last):
  File "", line 1, in 
  File "/home/yuki/caffe/python/caffe/__init__.py", line 1, in 
    from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver
  File "/home/yuki/caffe/python/caffe/pycaffe.py", line 13, in 
    from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \
ImportError: /home/yuki/.pyenv/versions/anaconda-2.1.0/bin/../lib/libm.so.6: version `GLIBC_2.15' not found (required by /usr/lib/x86_64-linux-gnu/libxvidcore.so.4)

のようなエラーが出てきた時は,とりあえず

/home/yuki/.pyenv/versions/anaconda-2.1.0/bin/../lib/libm.so.6

このファイルを削除すれば一応ライブラリを読み込むようになります(完全に解決したわけではない)

caffeの環境構築はやっぱり難しい(´・ω・`)

chainer

pip install chainer

これで終わり!!いいですねぇ

Tensorflow

なぜかライブラリ読み込まれていず苦戦なう
TensorFlow-Download and Setup

sudo apt-get install python-pip python-dev
#CPU
sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.7.1-cp27-none-linux_x86_64.whl
8
7
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
8
7