POSTED BY
2024-01-13

GPUがしっかり認識されたTensorFlow2でmodel.fitを行おうとした時に表題のエラー。

当方環境では、以下で解決。

Windowsの場合

一旦Anaconda Prompt、Jupyter Notebook等Python関係のプログラムを終了する。

コントロールパネル→システムとセキュリティ→システム→システムの詳細設定→環境変数

システム環境変数→新規

変数名：TF_FORCE_GPU_ALLOW_GROWTH
変数値：true

すべてＯＫで保存。当該エラーのコードを再実行したら今度はうまくいった。

Linuxの場合

bash系

export TF_FORCE_GPU_ALLOW_GROWTH=true

tcsh系

setenv TF_FORCE_GPU_ALLOW_GROWTH true

初期設定ファイル.bashrc, .cshrcなどに記述しておくとよい。