【Python】Error in PredictCost()のエラー

AnacondaでTensorFlow GPU環境を構築して事前学習済みモデルを利用しようとしたところ、エラーが出ました。

エラー内容

2022-03-08 20:43:00.279799: W tensorflow/core/grappler/costs/op_level_cost_estimator.cc:690] Error in PredictCost() for the op: op: "CropAndResize"

このようなエラーメッセージが出ました。

原因

どうやらTensorflowのバージョンに合うCUDAとcuDNNをインストールしてなかったのが原因っぽいです。

https://www.tensorflow.org/install/source_windows#gpu

上記のサイトからTensorflowに合うバージョンを探しました。

tensorflow_gpu-2.6.0をインストールしていたので、CUDAは11.2、cuDNNは8.1をインストールすれば良いようです。

バージョンを気にせずインストールしていたのでCUDAのアンインストールとか苦労しました。
私の場合はこれでエラーが消えて無事に動くようになりました。

PythonPyCharm,Python