运行 Darknet YOLOv4 在 Google Colab 上出现 CUDA 错误

CUDA error when running Darknet YOLOv4 on Google Colab

我认为 Colab 或 Cuda 最近的更新正在抛弃 YOLOv4 模型。我以前构建过这个笔记本并用它训练了一个完整的模型,没有任何问题,但是现在当我 运行 完全相同的代码,没有任何更改时,我遇到了这个问题:

 672 x 672 
 try to allocate additional workspace_size = 65.03 MB 
 CUDA allocate done! 
Loaded: 2.962808 seconds - performance bottleneck on CPU or Disk HDD/SSD

 CUDA Error Prev: an illegal memory access was encountered
Darknet error location: ./src/dark_cuda.c, check_error, line #81
CUDA Error Prev: an illegal memory access was encountered: File exists

还有其他人遇到过这个问题吗?它似乎是最近的,因为到目前为止还没有人能够弄清楚。这就是我 运行 设置我的暗网的方式:

!./darknet detector train data/obj.data cfg/custom-yolov4-detector.cfg yolov4.conv.137 -dont_show -map -clear

似乎 Google Colab 减少了免费笔记本允许的内存。

尝试更改配置中的批量大小和细分。

batch=32 和 subdivisions=8 为我解决了这个问题。