facebookresearch/maskrcnn-benchmark

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

Aberta

#325 aberto em 8 de jan. de 2019

Ver no GitHubGuia do colaborador
 (10 comentários) (0 reação) (0 responsável)Python (2.574 forks)batch import
awaiting responsehelp wanted

Métricas do repositório

Stars
 (9.161 estrelas)
Métricas de merge de PR
 (Métricas PR pendentes)

Description

❓ 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?

Guia do colaborador

Direção de pesquisa
Verifique a visibilidade da GPU do container Docker com 'nvidia smi'. Certifique se de que CUDA VISIBLE DEVICES inclua ambas as GPUs. Verifique a inicialização do grupo de processos distribuídos e o uso do backend NCCL.
Pilha de tecnologia
pythonpytorch
Domain
machine learningbackend
Tipo Issue
Bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
4
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Antigo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PyTorchDocker
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
30