今でROCm-Pytorch-Dockerの環境構築は毎回Buildしてあげないと駄目だったのがBuild無用になっていた
今までAMDなGPUでPytorchを動かそうとするとそれなりに手間がかかっていたので嬉しい.

https://qiita.com/T_keigo_wwk/items/1a525768ad7585bda34f
Pytorch-ROCm DockerでPytorch-ROCmをビルドしてみる(ROCm2.4編）

この時点ではbuildを毎回しないと構築不能だったようだが現在はpip3にプレインストールされるようになったみたいなのでROCｍ-Dockerを動かせる環境さえ立ち上げてしまえば即運用可能になると思われる.

動作検証そのものは特にまだしていないのでこれからコミュニティによる動作検証やRTGチームでの最適化が必要だと思われる.

今回確認したのはこのDocker image

 sudo docker pull rocm/pytorch:rocm2.7_ubuntu16.04_py3.6_pytorch

sudo   docker run --name pytroch_rocm -it  --privileged --rm --device=/dev/kfd --device=/dev/dri --group-add video rocm/pytorch:rocm2.7_ubuntu16.04_py3.6_pytorch

pip3 listでインストールされているか確認する.

# pip3 list
Package            Version               
------------------ ----------------------
absl-py            0.7.1                 
atomicwrites       1.3.0                 
attrs              19.1.0                
audioread          2.1.8                 
backcall           0.1.0                 
bleach             3.1.0                 
cffi               1.12.3                
Click              7.0                   
cycler             0.10.0                
decorator          4.4.0                 
defusedxml         0.6.0                 
entrypoints        0.3                   
future             0.17.1                
grpcio             1.22.0                
hypothesis         4.32.3                
imageio            2.5.0                 
importlib-metadata 0.19                  
ipykernel          5.1.2                 
ipython            7.7.0                 
ipython-genutils   0.2.0                 
ipywidgets         7.5.1                 
jedi               0.15.1                
Jinja2             2.10.1                
joblib             0.13.2                
jsonschema         3.0.2                 
jupyter            1.0.0                 
jupyter-client     5.3.1                 
jupyter-console    6.0.0                 
jupyter-core       4.5.0                 
kiwisolver         1.1.0                 
librosa            0.7.0                 
llvmlite           0.29.0                
Markdown           3.1.1                 
MarkupSafe         1.1.1                 
matplotlib         3.1.1                 
mistune            0.8.4                 
mock               3.0.5                 
more-itertools     7.2.0                 
mypy               0.720                 
mypy-extensions    0.4.1                 
nbconvert          5.6.0                 
nbformat           4.4.0                 
networkx           2.0                   
ninja              1.9.0.post1           
notebook           6.0.0                 
numba              0.45.1                
numpy              1.17.0                
packaging          19.1                  
pandocfilters      1.4.2                 
parso              0.5.1                 
pexpect            4.7.0                 
pickleshare        0.7.5                 
Pillow             6.1.0                 
pip                19.2.2                
pluggy             0.12.0                
prometheus-client  0.7.1                 
prompt-toolkit     2.0.9                 
protobuf           3.9.1                 
psutil             5.6.3                 
ptyprocess         0.6.0                 
py                 1.8.0                 
pycparser          2.19                  
pycurl             7.43.0                
Pygments           2.4.2                 
pygobject          3.20.0                
pyparsing          2.4.2                 
pyrsistent         0.15.4                
pytest             5.0.1                 
python-apt         1.1.0b1+ubuntu0.16.4.5
python-dateutil    2.8.0                 
PyWavelets         1.0.3                 
PyYAML             5.1.2                 
pyzmq              18.1.0                
qtconsole          4.5.2                 
resampy            0.2.1                 
scikit-image       0.15.0                
scikit-learn       0.21.3                
scipy              0.19.1                
Send2Trash         1.5.0                 
setuptools         41.1.0                
six                1.12.0                
SoundFile          0.10.2                
tabulate           0.8.3                 
tb-nightly         1.15.0a20190814       
terminado          0.8.2                 
testpath           0.4.2                 
torch              1.2.0a0+eddebaf       
torchvision        0.3.0a0+2b73a48       
tornado            6.0.3                 
traitlets          4.3.2                 
typed-ast          1.4.0                 
typing             3.7.4                 
typing-extensions  3.7.4                 
virtualenv         16.7.2                
wcwidth            0.1.7                 
webencodings       0.5.1                 
Werkzeug           0.15.5                
wheel              0.33.4                
widgetsnbextension 3.5.1                 
zipp               0.5.2

torch 1.2.0a0+eddebaf

ちゃんと追加されている

ここでは環境構築の手順だけ残しておくのですが今後動作検証とかするつもりです.

追記(2019/10/01)　python関連で気になることがあった

rocm2.7_ubuntu16.04_py3.6_pytorch のイメージだがなんか見てたら

# python3 -V
Python 3.5.2

python3.5がインストールされている。。。なぜか

あと

 import numpy
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'numpy'

pip3でinstall済みのはずなのにimportが出来ないなど諸問題があった
他のiamgeなどでも試してみたい

追記

pip3でimportできない問題が発生しているのは

python3 ./hoge.py

で実行しているときで

python3.5 (Dockerコンテナによっては3.6だったりするのでimage名を確認すること) ./hoge.py

だとimportできる、留意して使いたい.