运行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)
###############
即可解决问题