cvl-robot's diary

研究ノート メモメモ https://github.com/dotchang/

メモ:Windowsで使うKerasのバックエンドのtensorflow-gpuを好きなCUDAとPythonのバージョンで使う方法

pip install tensorflow-gpuでは、古いバージョンのCUDAやcuDNNしか対応してくれないので使えません。pythonのバージョンも限定されます。

tensorflowとkerasのインストール

新しいバージョンのCUDAやPythonを使いたい場合は、下記のサイトを参考に、
qiita.com
おそらく非公式のここから、欲しいバージョンに対応したwhlをダウンロードしてきます。
github.com
インストールは、例えばtensorflow_gpu-1.10.0-cp36-cp36m-win_amd64.whlなら

pip install tensorflow_gpu-1.10.0-cp36-cp36m-win_amd64.whl

GPU対応の確認

このtensorflow_self_check.pyをGoogle Chromeで開いてテキストファイルにコピペして、拡張子を.pyに変更して実行します。
TensorFlow on Windows self-check · GitHub