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

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)
相关文章
相关标签/搜索