如何使用 2 个 GPU 来训练同一个 yolov3 模型?
作者: thimabru1010创建于 2019年9月30日更新于 2025年8月10日
我在 PC 上训练 yolov3 遇到问题,我的 PC 配备了 2 个 8 GB 的 GPU。问题在于,我只能将 Batch 设置为 64,Subdivisions 设置为 64。对于 Subdivisions 的其他值(32、16、8)都会产生错误: CUDA 内存不足。问题在于,我有 2 个 GPU,而程序并没有识别多余的 GPU 并分配给另一个 GPU,即使设置了 -gpus 0,1 并使用 CUDA 编译。我想降低 Subdivisions 以获得更高的准确性。如何同时使用两个 GPU 来训练同一个 yolov3 模型?
内容来源: pjreddie/darknet