Windows10上でKeras使ってGPU演算 (CUDA v9.2)

f:id:thirty_eesti:20180806040806j:image

 

授業で出番がくるだろうからWindows10でKerasを動かすやりかたを試してみた。GPU演算。

 

Visual Studio 2017(C++, SDK)

CUDA(v9.2)

cudnn(v9.2)

Anaconda(Python 3.5.5)

Windows10 Pro

GeForce GTX 1070

 

授業で気をつけたいことは、GUIでTensorflow-gpuインストールしないでCUIのAnaconda Promptでインストールしてねーってところ。

 

システム環境変数は勝手にプログラム側がパス通してくれてたし、こちら側が手動で環境変数を追加する必要はない。

 

不安な点はcudnnのファイル置き換えがフォルダごとなのかフォルダ直下なのかはっきりしてないくらい。

 

GPU環境(Anaconda関係なしに)が整ってるかはコマンドプロンプトでnvccと打てば確認できる。nvcc fatalと返ってくればよい。

 

Tensorflow上でGPU使ってるかの確認はSpyder上でTensorflow呼び出してなにかしらのコードを実行すればよい。タスクマネージャーでGPU使用が認められればOK。

 

 

参考リンク

qiita.com

 

liaoyuan.hatenablog.jp

 

qiita.com

 

-----追記(2018.10.30)-----

研究室のPCでやる場合+OpenCV使う場合は、Adminで作業したほうがよいです。権限の問題でimportできない場合がある…。あと、仮想環境はPython3.7は現時点(2018.10.30)では使わないでください。Tensorflow-gpuがインストールできない…。あと研究室のPCはCUDAの9.0だと思います。9.2じゃないです。