search

uni-medical / SAM-Med3D

SAM-Med3D: An Efficient General-purpose Promptable Segmentation Model for 3D Volumetric Medical Image
Apache License 2.0
427 stars 56 forks source link

Multi-GPU is not working #30

Closed harpergith closed 6 months ago

harpergith commented 6 months ago

Hello,

I tried to use multiple GPUs by setting --multi_gpu, however, only GPU 0 was used during training according to the outputs of nvidia-smi command.