-
建议不要同时安装tensorflow CPU版本和GPU版本,否则可能由于版本冲突导致keras训练模型时无法使用到GPU
-
tensorflow-gpu版本建议不要使用高版本,使用1.13.1即可甚至还可以降低版本,同时需要注意CUDA版本需要对应
-
报错信息:E tensorflow/stream_executor/cuda/cuda_dnn.cc:334] Could not create cudnn handle: CUDNN_STATUS_INTER
解决方法:
加入下列代码:
import tensorflow as tf
from tensorflow.compat.v1 import ConfigProto
config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)
或者是:
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)