facebookresearch/maskrcnn-benchmark

why only one gpu be used when I launch mutil gpu train?

开放

#325 创建于 2019年1月8日

 (10 条评论) (0 个反应) (0 位负责人)Python (2,574 个派生)batch import
awaiting responsehelp wanted

仓库指标

星标
 (9,161 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

❓ Questions and Help

when I launch multi gpu train use this command:

python -m torch.distributed.launch --nproc_per_node=2 /path_to_maskrcnn_benchmark/tools/train_net.py --config-file "path/to/config/file.yaml"

and it's run in docker container env, there are have two p40 in docker, but it seems that only one GPU is being used. image

and there are actually two processes running image

What could be the reason?

贡献者指南