機械学習

DockerでCaffeを実行する

caffeのインストール

caffeのインストールはとても大変です。anacondaとかpipでインストールするだけではうまくいきません。

Docker

Dockerを利用してcaffeをインストールします。import caffeまで非常に簡単にインストールできます。

# imageからcontainerを作成する
$ docker run -ti bvlc/caffe:cpu caffe --version
# containerを起動する
$ docker start コンテナ名

1分以内に環境の構築が完了します。

# 以下のコマンドで、Docker内にいなくても操作できる
$ docker run -ti bvlc/caffe:cpu ipython
# docker内にアクセスできます
$ docker attach コンテナ名

jupyter notebook

# Dockerにアクセスする
$ docker exec -it コンテナ名 /bin/bash

anacondaは入っていなかったがpipが入っていたので、pipjupyterをインストールしました。

$ pip install jupyter

そして、起動させます。起動させる時はipアドレスの指定をして、Docker上ではrootでの起動になるので--allow-rootオプションをつけます。

$ jupyter notebook --ip=127.0.0.1 --allow-root

port指定

# イメージをpull
$ docker pull kyamagu/caffe
# port8888を通す
$ docker run -it -p 8888:8888 kyamagu/caffe

参考

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です