【问题标题】:What is "cudnn_status_internal_error" meaning Tensorflow pipeline?什么是“cudnn_status_internal_error”,意思是 Tensorflow 管道?
【发布时间】:2019-08-30 18:48:30
【问题描述】:

如果我们使用 TensorFlow 模型 API 启动 object_detection 项目,则会收到错误“cudnn_status_internal_error”。

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

如何解决?

【问题讨论】:

    标签: tensorflow object-detection object-detection-api cudnn


    【解决方案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)
    
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2020-10-08
      • 2013-11-26
      • 1970-01-01
      • 2018-01-12
      • 2011-09-05
      相关资源
      最近更新 更多