tensorflow调用GPU时报错:
再模型调用前加上如下代码:
config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)
自己的代码是:
config = tf.ConfigProto(allow_soft_placement=True)
sess = tf.Session(config=config)
E tensorflow/stream_executor/cuda/cuda_dnn.cc:371] could not create cudnn handle: CUDNN_STATUS_INTE
最新推荐文章于 2023-01-04 21:58:14 发布