This is the code for the PyTorch extension for 3D Slicer.

Its main function is to install PyTorch inside Slicer. The latest version compatible with the installed drivers will be selected automatically. CI tests are run nightly.

PyTorch can be installed opening the PyTorch Utils module and clicking on the button, or programmatically:

import PyTorchUtils
torch = PyTorchUtils.PyTorchUtilsLogic().torch  # will be installed if necessary
tensor = torch.rand(50, 60)

Please use a Preview Release of 3D Slicer (currently 4.13, not 4.11).

Unit tests are run nightly and the results can be checked on CDash.

Here's a diagram of the integration between PyTorch and Slicer: