运行tensorflow深度学习代码,出现卡死界面、或报Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR错的解决方法

运行tesnsorflow较为复杂的深度学习模型时,容易出现卡死界面的情况。
如果是单显卡(cpu型号尾部有f),可安装opengl并重启后,再运行代码。
如果出现Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR报错,则是显存不够导致的,加入

#用于限制显存使用
config = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True))
sess = tf.compat.v1.Session(config=config)
###############

即可解决问题

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