YOLO-V3-IOU icon indicating copy to clipboard operation
YOLO-V3-IOU copied to clipboard

GPU

Open zhonganzuzhihu opened this issue 5 years ago • 0 comments

这个代码怎么限制GPU使用情况,我用 config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.1 session = tf.Session(config=config)

self.sess = K.get_session(session) 这样是错误的,作者大神有什么方法限制显存么

zhonganzuzhihu avatar Jul 17 '19 09:07 zhonganzuzhihu