Use AMD ROCm for TotalSegmentator

Is there a way to use AMD cards with ROCM? I am thinking of supported cards with the AUR (en) - python-pytorch-rocm ( package or card specific wheels like xuhuisheng/rocm-gfx803 (

Looks like auto-installation with the light the torch is only for the CUDA based GPUs. light-the-torch ยท PyPI

Torch already supports those GPUs, so you should be able to manually install those.

Light-the-torch supports CPU, CUDA, ROCm backends. Therefore, all ROCm versions that pytorch supports should work well in Slicer.

1 Like