search

alievk / npbg

Neural Point-Based Graphics
MIT License
324 stars 51 forks source link

Multi-GPU support for training? #36

Closed Yurri-hub closed 1 year ago

Yurri-hub commented 1 year ago

I try to train a large-scale scene with large number of point clouds. But I received a CUDA out of memory error at the start of training.

Is there any way to train the model on Multi-GPUs?

seva100 commented 1 year ago

Hi @Yurri-hub, yes, you can specify multigpu: True in train config or --multigpu in command line when running python train.py. The indices of the GPUs used can be specified via CUDA_VISIBLE_DEVICES="0,1,5" python train.py ..., where indices should be specified instead of "0,1,5". Also make sure to set the batch_size accordingly in order to avoid memory issues.

Yurri-hub commented 1 year ago

Thank you for your reply. It helps a lot.