How to specify which gpu to use for training

Kai-46 / nerfplusplus

improves over nerf in 360 capture of unbounded scenes

BSD 2-Clause "Simplified" License

911 stars 101 forks source link

Open yhl1010 opened 2 years ago

yhl1010 commented 2 years ago

If I want to train only with the num'2' and num'3' gpus, how should i change the code? thank you very much!

zParquet commented 2 years ago

In linux command line, train with "CUDA_VISIBLE_DEVICES=2,3 python ddp_train_nerf.py xxxxxx "