不明:たたみ込みアルゴリズムの取得に失敗しました。これはおそらく、cuDNNが初期化に失敗したためです...

実行中のコードでtensorflow 1.9.0が必要だったために上記の問題が発生しましたが、私は1.14.0でした。バージョンをダウングレードするために多くの方法をチェックしたので、後で次の方法を採用し、それをコードの先頭に置いて解決しました。

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

 

おすすめ

転載: blog.csdn.net/weixin_45191152/article/details/103715785