2017年10月08日
Chainer のインストールに手こずる
- 2017-10-08 (Sun)
- ソフトウェア
Chainer というより CuPy のインストールに四苦八苦．最初に CUDA 9 を入れたらダメっぽいので CUDA 8 に入れ直してみたり，その後に cuDNN が有効になってないとか warning が出ているので cuDNN を入れて諸々して pip install --no-cache-dir -I で何度か再コンパイルするも cuDNN が全然認識されなかったり．結局 cuDNN は無くても動くことがわかったので諦めた．
んで，Chainr のサンプル train_mnist.py を time で測ってみたら CPU（8コアx2）で 6分，GPU（1080）で 50秒．GPU のが 7倍くらい速い．cuDNN 有効にしたらもっと速くなるのだろうか？
とりあえずこれで DL で遊べる環境が出来たので，何かネタ学習をやらせてみようかね．
- Comments: 0
- TrackBack (Close): -