keras使用GPU加速：Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

kerasGPU训练模型问题：Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
首先找到tensorflow session所在的目录
/anaconda/envs/zh/lib/python3.7/site-packages/tensorflow/python/client/
编辑 vi session.py

直接输入 :set number

shift +g 直接进入最后一行 大概在1570行
确定好修改 session 这个类初始化函数

config=config_pb2.ConfigProto(gpu_options=config_pb2.GPUOptions(allow_growth=True))

这行代码必须放在以下两行代码中间 ②处

class Session(BaseSession):
…①
def init(self, target=’’, graph=None, config=None):
…②
super(Session, self).init(target, graph, config=config)

保存退出，问题解决

修改前

修改后