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)
保存退出,问题解决