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)

你可能感兴趣的:(tensorflow,神经网络,深度学习)