how to use multi-gpus training?

sconlyshootery / FeatDepth

This is the offical codes for the methods described in the "Feature-metric Loss for Self-supervised Learning of Depth and Egomotion".

MIT License

247 stars 28 forks source link

how to use multi-gpus training? #88

Closed 760677482 closed 2 years ago

760677482 commented 2 years ago

i training by ‘ CUDA_VISIBLE_DEVICES=0,2 python -m torch.distributed.launch --master_port=9900 --nproc_per_node=2 train.py --launcher pytorch -work_dir ... --config ...’ Then displays the error :subprocess.CalledProcessError: Command '['.../envs/bin/python', '-u', 'train.py', '--local_rank=1']' returned non-zero exit status 1.