nnInteractive with GPU 5090

Hi -

I am interested to use nnInteractive extension on Nvidia 5090 series laptop. Unfortunately, whenever I try to use Totalsegmentator or nnInteractive, it re-installs an older version of pytorch which is incompatible with Cuda 13.0. Could you please help?

If you are using the PyTorch extension to install the torch, I think it doesn’t bring anything newer than cuda 12.6 (at least that has been my experience).

If you need torch with cuda 13.0, you probably need to manually install it using the command from torch site:

pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu130

So perhaps something like this in Slicer’ python:

slicer.util.pip_install(['torch', 'torchvision', '--index-url', 'https://download.pytorch.org/whl/cu130'])