YOLOX训练模型多GPU配置

问题遇到的现象和发生背景

yolox训练模型时,只能用一个GPU

用代码块功能插入代码,请勿粘贴截图

一共有三个地方可以设置
train.py
parser.add_argument(
"-d", "--devices", default=4, type=int, help="device for training"

launch.py
def launch(
main_func,
num_gpus_per_machine,
num_machines=4,
machine_rank=0,
backend="nccl",
dist_url=None,
args=(),
timeout=DEFAULT_TIMEOUT,

开始训练指令
python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 0 -d 4 -b 16 --fp16 -o -c weights/yolox_s.pth

运行结果及报错内容

各种奇怪报错指令

我的解答思路和尝试过的方法

没有办法了

我想要达到的结果

能够实现多GPU训练(4GPU)

你好,我是有问必答小助手,非常抱歉,本次您提出的有问必答问题,技术专家团超时未为您做出解答


本次提问扣除的有问必答次数,已经为您补发到账户,我们后续会持续优化,扩大我们的服务范围,为您带来更好地服务。