{cudnn_status_internal_error“是什么意思Tensorflow管道？

Question

如果使用TensorFlow Model API启动object_detection项目，则会收到错误“ cudnn_status_internal_error”。

我正在使用TensorFlow对象检测管道来训练模型。

如何解决？

Answer 1

就我而言，这是因为GPU资源不足。

要在管道中解决此问题，请在“ model_main.py”中添加一段代码。

  session_config = tf.ConfigProto()
  #session_config.gpu_options.allow_growth = True
  session_config.gpu_options.per_process_gpu_memory_fraction = 0.8

  config = tf.estimator.RunConfig(model_dir=FLAGS.model_dir, session_config=session_config)