【发布时间】:2019-08-30 18:48:30
【问题描述】:
如果我们使用 TensorFlow 模型 API 启动 object_detection 项目,则会收到错误“cudnn_status_internal_error”。
我正在使用 TensorFlow 对象检测管道来训练模型。
如何解决?
【问题讨论】:
标签: tensorflow object-detection object-detection-api cudnn
如果我们使用 TensorFlow 模型 API 启动 object_detection 项目,则会收到错误“cudnn_status_internal_error”。
我正在使用 TensorFlow 对象检测管道来训练模型。
如何解决?
【问题讨论】:
标签: tensorflow object-detection object-detection-api cudnn
就我而言,是因为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)
【讨论】: