はじめに
tensorflowを使いたいなーと思ってCudnnをインストールしたら、また新しいエラーが出てきました。結構困ったのでメモります。
エラー内容
GPU自体はtensorflowで読み込めているのだが、実行するとなんだかエラーが発生する。エラー内容は以下。
error
...:...: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
...:...: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
Traceback ...
...
解決方法
以下のコマンドを実行すると無事解決した。
$ export TF_FORCE_GPU_ALLOW_GROWTH=true