UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initiali

566 ワード

最近、構成環境が完了した後、モデルを実行する際にエラーが発生した.ほとんどのブロガーはバージョン構成の問題のため、tensorflowバージョンが1.14または1.12で1.9に降格することを提案した.また、多くのブロガーは再起動の大法を採用し、午後を振り回した.現在、個人が見つけた比較的良い解決策は、個人バージョンがpython 3である.7+tensorflow1.13,
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.99
with tf.Session(config=config) as sess:
......... 

个人的な推测では、GPUは実行开始时にGPUを占有するプログラムがあると思っているか、あなたが现在実行しているプログラムが大量のGPUを占有していると思っている可能性があります.あるノードに実行する时、GPUが完全に占有されていると思っています.皆さんに役に立つことを望んでいます.