Repository navigation
Add torchvulkan as fallback when no cuda is present #26
Description
context https://discourse.slicer.org/t/use-torchvulkan-to-enable-gpu-segmentations-in-totalsegmentator/48679
You can use vulkan from https://github.com/zainsharief/torchvulkan in addition to torch to define a vulkan device as fallback when no cuda.
This would allow using torch on any GPU, not just nvidia
Activity
lassoan commentedon Nov 3, 2026
Please submit a PR that adds this feature to this extension in a way that it works for you. The critical part is to automatically detect the case when cuda is not available but vulkan is, because only in that case torchvulkan should be attempted to be installed. We need to be careful with installing torchvulkan: it should only be attempted to install torchvulkan only if needed, because it is not stable yet and it is available only for certain pytorch versions.
BishopWolf commentedon Nov 3, 2026
I will try to add a PR this weekend