Is there a way to use AMD cards with ROCM? I am thinking of supported cards with the AUR (en) - python-pytorch-rocm (archlinux.org) package or card specific wheels like xuhuisheng/rocm-gfx803 (github.com)
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