nerfplusplus icon indicating copy to clipboard operation
nerfplusplus copied to clipboard

Published 20 hours ago • verified publisher icon Kai-46

How to specify which gpu to use for training

Open yhl1010 opened this issue 3 years ago • 1 comments

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

yhl1010 avatar Feb 22 '22 09:02 yhl1010

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

wen-yuan-zhang avatar Mar 13 '22 09:03 wen-yuan-zhang