如何利用多CPU来训练YOLO？

Question

我可以访问没有 GPU 的大型 CPU 集群。是否可以通过多个CPU节点之间的并行来加速YOLO训练？
文档说

device

参数指定用于训练的计算设备：单个 GPU (device=0)、多个 GPU (device=0,1)、CPU (device=cpu) 或 Apple 芯片的 MPS (设备=mps）。 多个 CPU 怎么样？

Answer 1

您可以使用

torch.set_num_threads(int)

（docs）来控制pytorch使用多少个CPU进程来执行操作。