【无标题】

Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

解决办法:在开头加入以下代码

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.5)
config = tf.ConfigProto(gpu_options=gpu_options)
config.gpu_options.allow_growth = True
session = tf.Session(config=config)
 

你可能感兴趣的:(tensorflow,深度学习,人工智能)