GPUが認識されているのに、使えない。

Question

PythonでGPUを使った計算がしたいのですが、エラーが出てしまいます。

環境は以下の通りです。
GPU:RTX2060
CUDA Toolkit 10.0
cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.0
tensorflow-gpu 2.0.0
Python3.6.10(Python3.7だとできないらしい)
jupiter notebook 6.0.3
anaconda3
windows10 64bit

Answer 1

おそらく個別にダウンロードしたCUDA Toolkit/cuDNNと、Anacondaによってインストールされたcudatoolkitおよびcudnnが衝突してトラブルを起こしています。

一番簡単な解決方法は個別にインストールしたCUDA ToolkitとcuDNNをアンインストールすることです。

Anacondaはtensorflow-gpuをインストールする際に、依存関係のあるcudatoolkitとcudnnを一緒にインストールしてくれますのでリンク先にあるようなインストールは不要です。

Comments

[rest]

回答ありがとうございます。
個別にインストールしたcudatoolkitとcudnnをアンインストールしました。
さらに、新しい仮想環境を構築しtensorflow-gpu2.0.0をインストールしたところ、同時にCUDA10.0.130とcuDnn7.6.5がインストールされたことを確認しました。
また、TensorFlowがGPUを認識しているか確認したところ、認識されていました。
しかし、keras2.3.1を使ったプログラムを回してみたところ、エラーが出てしまいました。

UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

GPUが使われるようになるには、どのようにすれば良いかアドバイスを頂けませんか？

[yymmt]

環境変数が残っていませんか?それを削除しないとそちらが優先されてしまいます。

[rest]

ありがとうございます。
環境変数が残っていたので、それを削除したら、今度はResourceExhaustedErrorが出てしまいました。
他の環境では実行できるのですが、何故かgpuの使おうとしているこの環境だけできません。

[yymmt]

GPUのメモリが足りていないように思います。batch_sizeを小さくするなどの対応が必要です。

[rest]

PCを再起動してみたところ、エラーが出ないようになりました。
どうやら、環境変数を変更したので、一度再起動した方が良かったみたいです。
しかし、エラーは出ないものの、依然としてGPUの使用率は0%です。
他に、足りてない処理の心当たりはありますか？

[yymmt]

GPU使用率の確認はどのように行っていますか?タスクマネージャで専用GPUの使用率が0%という状況でしょうか?

[rest]

タスクマネージャから見て、GPU使用率は0%です。
また、NZXT_CAMから見てもGPUの使用率は0%です。
それに対して、CPUの使用率は95%前後です。

[yymmt]

問題点を切り分けたいので、まずはTensorflowのみでGPUのチェックを順次行うのはどうでしょうか。
https://www.tensorflow.org/guide/gpu

[rest]

返信ありがとうございます。
問題点がだいぶ絞り込めてきました。
どうやら、GPU自体は使えているようで、特定のプログラムにのみGPUが使えていない状況でした。
最初の質問の趣旨とは違う内容になってしまいますので、違う投稿で改めて投稿させていただきます。
ありがとうございました。