search

bubbliiiing / yolo3-pytorch

这是一个yolo3-pytorch的源码，可以用于训练自己的模型。
MIT License
2.01k stars 585 forks source link

请问如何防止模型在服务器训练的时候占用所有的GPU？ #165

Closed tang-mouren closed 1 year ago

tang-mouren commented 1 year ago

我在train.py distributed=false的情况下，进行训练的时候会占用服务器的所有GPU。请问我应当如何设置占用的GPU数量上限呢？