アプリケーション開発ポータルサイト
ServerNote.NET
カテゴリー【PythonWindowsDebian
Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILEDエラーの対処法
POSTED BY
2023-01-13

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

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

Windowsの場合

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

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

システム環境変数→新規

変数名:TF_FORCE_GPU_ALLOW_GROWTH
変数値:true

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

Linuxの場合

bash系

export TF_FORCE_GPU_ALLOW_GROWTH=true

tcsh系

setenv TF_FORCE_GPU_ALLOW_GROWTH true

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

※本記事は当サイト管理人の個人的な備忘録です。本記事の参照又は付随ソースコード利用後にいかなる損害が発生しても当サイト及び管理人は一切責任を負いません。
※本記事内容の無断転載を禁じます。
【WEBMASTER/管理人】
自営業プログラマーです。お仕事ください!
ご連絡は以下アドレスまでお願いします★

【キーワード検索】