tensorflow中GPU相关设置解决显存不足

最新推荐文章于 2023-09-30 13:18:12 发布

AICVer

最新推荐文章于 2023-09-30 13:18:12 发布

阅读量1.5w

点赞数 1

本文链接：https://blog.csdn.net/u011489887/article/details/80343698

版权

1.显存不足时报错如下：

E tensorflow/stream_executor/cuda/cuda_dnn.cc:359] could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

E tensorflow/stream_executor/cuda/cuda_dnn.cc:326] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM

2018-03-12 10:55:56.078991: F tensorflow/core/kernels/conv_ops.cc:671] Check failed: stream->parent()->GetConvolveAlgorithms(&algorithms) Aborted (core dumped)

Process finished with exit code -1073740791 (0xC0000409)

解决如下：

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=FLAGS.gpu_memory_fraction)
config = tf.ConfigProto(log_device_placement=False,gpu_options=gpu_options)
saver = tf.train.Saver(tf.global_variables())
summary_op = tf.summary.merge_all()
init = tf.initialize_all_variables()
sess = tf.Session(config=config)