PyTorch を nvidia-docker で動かしてみたかったので、やってみた。

ホスト OS は，Ubuntu16.04 です．

PyTorch の Dockerfile

PytTorch の Docker 環境は，公式のリポジトリにあるので，まずは git clone

.sh

git clone https://github.com/pytorch/pytorch

これは，Opencv をインストールするために必要な作業で，Opencv3 が，python3.6 に対応していない&上記の Dockerfile は，python3.6 をインストールするように設定されている．

ENV PYTHON_VERSION=3.6

を

ENV PYTHON_VERSION=3.5.2

こう。

.sh

pip install opencv-python

普段なら上記のようにしてインストールするのですが，今回はこれではインストール後正常に動作しなかったので，conda からインストールします．

conda install --name pytorch-py$PYTHON_VERSION -c https://conda.anaconda.org/menpo opencv3

公式の Dockerfile では，miniconda のインストール後に，上記で指定した PYTHON_VESHRSION の仮想環境を作って，そこで Pytorch が動くようになっているので，その仮想環境上に Opencv をインストールするようにしています．

これでめでたく PyTorch と opencv が動く Docker 環境が構築できます．
Dockerfile 全体は下記の通りです．