how to use the specifed gpu

HRNet / HRNet-Object-Detection

Object detection with multi-level representations generated from deep high-resolution representation learning (HRNetV2h). This is an official implementation for our TPAMI paper "Deep High-Resolution Representation Learning for Visual Recognition". https://arxiv.org/abs/1908.07919

Apache License 2.0

643 stars 98 forks source link

how to use the specifed gpu #14

Closed zkunob closed 4 years ago

zkunob commented 5 years ago

there are 8 gpus, the id is 0,1,2,3,4,5,6,7, I ues the CUDA_VISIBLE_DEVICES=4,5,6,7 ,but it doesn't work

wondervictor commented 5 years ago

CUDA_VISIBLE_DEVICES=4,5,6,7 python -m torch.distributed.launch --nproc_per_node 4 tools/train.py works.

wondervictor commented 4 years ago

No more response. closed