CUDA：暗网yolov1上的内存不足错误

Question

我正在尝试在GPU Tesla K40上运行darknet YOLOv1代码。

几天前，makefile中的参数“ARCH = --gpu-architecture = compute_20 --gpu-code = compute_20”正常运行。

但是，今天我再做一次。 “make”工作但是在视频上运行YOLO的演示时出现以下错误

我尝试了不同的ARCH参数，例如compute_30,35,52 ..但没有任何工作。

可能的解决方案是什么？

注意：darknet yoloV2工作正常。我正在实施以下博客：http://guanghan.info/blog/en/my-works/train-yolo/

Answer 1

您是否尝试修改 cfg / tiny-yolo-voc.cfg ？

它应如下所示：

[net]
batch=64
subdivisions=64
width=416
height=416
channels=3
momentum=0.9
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1
...

诀窍是让批次等于细分。