search

CGuangyan-BIT / PointGPT

[NeurIPS 2023] PointGPT: Auto-regressively Generative Pre-training from Point Clouds
MIT License
179 stars 18 forks source link

Multi GPU pretraining #11

Closed HuguesTHOMAS closed 9 months ago

HuguesTHOMAS commented 9 months ago

Hi,

Which command should be used to start the pretraining on multiple GPUs? I guessed from looking at the code, that the parameter --launcher pytorch needs to be set. Is there something else to set like environment variables etc?

Best, Hugues

CGuangyan-BIT commented 9 months ago

You can directly specify multiple GPUs, for example, CUDA_VISIBLE_DEVICES=0,1,2,3