autonomousvision / monosdf

[NeurIPS'22] MonoSDF: Exploring Monocular Geometric Cues for Neural Implicit Surface Reconstruction
MIT License
573 stars 53 forks source link

multigpu training #27

Closed cv-lab-x closed 2 years ago

cv-lab-x commented 2 years ago

hi, thanks for your great work! i got some questions about the multi gpu training, i set the code as the readme CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7,8 python -m torch.distributed.launch --nproc_per_node 8 --nnodes=1 --node_rank=0 training/exp_runner.py --conf confs/tnt_highres_grids_courtroom.conf but it still uses only one gpu,

looking forward to your reply, thanks! @niujinshuchong