search

CGuangyan-BIT / PointGPT

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

Multi GPU pretraining #11

Closed HuguesTHOMAS closed 1 year ago

HuguesTHOMAS commented 1 year 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 1 year ago

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