search

meidachen / STPLS3D

🔥 Synthetic and real-world 2d/3d dataset for semantic and instance segmentation (BMVC 2022 Oral)
236 stars 20 forks source link

multi gpu #32

Open minimini-1 opened 1 year ago

minimini-1 commented 1 year ago

I saw that "import torch.distributed as dist" in HAIS's train.py code. Can I use a multi gpu when I train a HAIS model??

If not, could give me any advise how to execute multi gpu possible? Thank you.

meidachen commented 11 months ago

@jongmin4422 , Sorry for the late reply, but I didn't try to run multiple GPUs myself, you may want to check with the HAIS repo author for this.

Greatest-Marx commented 8 months ago

I saw that "import torch.distributed as dist" in HAIS's train.py code. Can I use a multi gpu when I train a HAIS model??

If not, could give me any advise how to execute multi gpu possible? Thank you.

Hello, have you successfully run HAIS using multiple GPUs?