Chainerを使って画像系の深層学習を実装して遊びます。
- macOS Sierra 10.12.5 (pycrayon server)
- Ubuntu 16.04 (GPU server)
- Docker nvidia-docker を用いた環境構築
crayon(pycrayon)
を用いたTensorBoard
による簡易的な可視化(試験運用)
以下の手順に従って各種環境を導入します
- Dockerのインストール
Docker公式サイトを見て頑張る。Macでの可視化、Ubuntuでのプログラム実行をそれぞれ行いたかったので、両方にDockerをインストール。 - nvidia-dockerの導入
参考HPを見て頑張る。他にもいろいろ情報はあるかと思います。今回は、GPUがあるUbuntuに導入。 chainer
,CUDA
,cuDNN
のインストール
バージョン管理や、割と面倒なCUDA
周りのインストール作業を楽に済ませるため、各種モジュールが含まれるDockerイメージを作成して、コンテナを作る。Dockerイメージはバージョンが合ったものが見つからなかったのでざっくり作成。crayon
,pycrayon
のインストール
どちらもインストールについては、このページを参照してインストール。