Windows10 + GTX1080Ti + Chainer

環境構築

GTX1080Tiでディープラーニングを加速させたい.それだけ.

 

1. GTX 1080TiをPCI Expressに繋ぐ

ちなみに電源は600Wにしました.

 

2.BIOSの設定を確認してビデオカードの優先度をPCI Express > オンボードにする

 

3.ディスプレイとGPUを繋ぐ.

要するにGPU経由でディスプレイを表示してみる.この段階ではドライバとかは必要ない.むしろドライバをインストールする際にデバイスを認識させる必要があるのだが, GPU経由でディスプレイを表示しないと認識されなかった.

 

4.GeForceのドライバをインストール

とりあえずいれれば入る.

 

5.Visual Studio 2015をインストール

カスタムインストールにしてVisual C++ にチェックを入れる

 

6.CUDA 9.0をインストール

8だとGTX1080Tiを認識してくれなかった.

 

7.cuDNN 7.0をインストール

インストールというかzipの中身をコピーするだけ.

Windows向けのセットアップガイドだと5.1とか多いけど, CUDA9にはcuDNN7からしか対応してない.

 

8.pip install cupy --no-cache -vvvvする

ここでNVIDIA Tools Extensionに関するWarningが出る場合は環境変数PATHとINCLUDEにパスを追記することで消せる.消さなくてもイケるかはわかりません.

追記するパスはC:¥Program Files¥NVIDIA Corporation¥NvToolsExt¥以下を確認して下さい.

その他にNCCLに関するWarningが出るが無視でOK.

バージョンは2系.

 

9.pip install chainerする

ここでコケたことはない.バージョンは3系.

 

 

NCCLに関するWarningは無視できる.

そしてLinuxをレコメンドされる.

github.com

 

 

とりあえずChainer公式のMNIST Exampleが秒間20イテレーションから160イテレーションになった.