object detection錯誤之Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

在trainer.py 中

session_config = tf.ConfigProto(allow_soft_placement=True,
                                    log_device_placement=False)
    

下添加：

session_config.gpu_options.allow_growth = True

即可