facebookresearch/maskrcnn-benchmark

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

Ouverte

#325 ouverte le 8 janv. 2019

Voir sur GitHubGuide contributeur
 (10 commentaires) (0 réaction) (0 personne assignée)Python (2 574 forks)batch import
awaiting responsehelp wanted

Métriques du dépôt

Stars
 (9 161 étoiles)
Métriques de merge PR
 (Métriques PR en attente)

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?

Guide contributeur

Direction de recherche
Vérifiez la visibilité du GPU du conteneur Docker avec 'nvidia smi'. Assurez vous que CUDA VISIBLE DEVICES inclut les deux GPU. Vérifiez l'initialisation du groupe de processus distribués et l'utilisation du backend NCCL.
Stack technique
pythonpytorch
Domaine
machine learningbackend
Type d'issue
Bug
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
4
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Ancienne
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
PyTorchDocker
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
30