在使用keras时报错显示tensorflow 1.4 could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR的解决办法

在代码中增加下面一些代码即可

import os
import tensorflow as tf
from keras.backend.tensorflow_backend import set_session

os.environ["CUDA_VISIBLE_DEVICES"] = "0"  # 指定GPU

config = tf.ConfigProto()
config.gpu_options.allow_growth = True  # 按需
set_session(tf.Session(config=config))