anacondaがcudnnを認識しない場合

前提

cudnn もCUDA入れてtensorflow = 2.1 をインストールして以下のコードを実行してGPUを認識してる確認した。

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

結果

CPUしか認識していない。これは困った

調査

conda list を実行して作った環境の中にcudnnが存在するか確認する。

今回私の環境ではcudnnの欄が存在しなかった。

解決策

conda install tensorflow-gpu をインストールする。

tensorflow = 2.1 をインストールしたら同時に対応するものだと思っていたがtensorflow-gpu を入れたら自動的にcudnnを認識してくれるようになった。