"cudnn_status_internal_error" 是什么意思？
What is "cudnn_status_internal_error" meaning Tensorflow pipeline?
如果我们使用 TensorFlow 模型 API 启动一个 object_detection 项目，我们会收到一个错误 "cudnn_status_internal_error"。
我正在使用 TensorFlow 对象检测管道来训练模型。
如何解决？
我的是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)
如果我们使用 TensorFlow 模型 API 启动一个 object_detection 项目，我们会收到一个错误 "cudnn_status_internal_error"。
我正在使用 TensorFlow 对象检测管道来训练模型。
如何解决？
我的是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)