2021-04-01 18:05:04.602841: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudn

原创于 2021-04-05 11:34:28 发布 · 1.4k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #神经网络 #深度学习

本文介绍了一种解决GPU显存不足的问题，通过设置允许显存在需要时增长的方式，避免了预分配全部显存导致的资源浪费。具体实现是通过TensorFlow的ConfigProto类进行配置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

GPU显存设置有问题，需要设置为仅在需要时申请显存。

import tensorflow as tf
config = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True))
sess = tf.compat.v1.Session(config=config)