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)
相關文章
相關標籤/搜索