“什么是”呢;cudnn“状态”内部“错误”;什么是Tensorflow管道?
如果我们使用TensorFlow模型API启动一个对象检测项目,我们会得到一个错误“cudnn\u status\u internal\u error” 我正在使用TensorFlow目标检测管道来训练模型“什么是”呢;cudnn“状态”内部“错误”;什么是Tensorflow管道?,tensorflow,object-detection,object-detection-api,cudnn,Tensorflow,Object Detection,Object Detection Api,Cudnn,如果我们使用TensorFlow模型API启动一个对象检测项目,我们会得到一个错误“cudnn\u status\u internal\u error” 我正在使用TensorFlow目标检测管道来训练模型 如何解决它?在我的例子中,这是因为GPU资源不够 要在管道中解决此问题,请在“model_main.py”中添加一段代码 session\u config=tf.ConfigProto() #session\u config.gpu\u options.allow\u growth=True
如何解决它?在我的例子中,这是因为GPU资源不够 要在管道中解决此问题,请在“model_main.py”中添加一段代码
session\u config=tf.ConfigProto()
#session\u config.gpu\u options.allow\u growth=True
session\u config.gpu\u options.per\u进程\u gpu\u内存\u分数=0.8
config=tf.estimator.RunConfig(model\u dir=FLAGS.model\u dir,session\u config=session\u config)