Can MONAI Auto3dSeg use metal (MPS) on M series processors running MacOS?

Support
1

I am running Auto3dSeg on my MacBook Pro with M3 Pro and wondering whether there is something I need to do to enable use of MTL for inferencing?

Thank you.

Gene Risi

2

The MPS (metal performance shaders) support will be possible once Slicer provides a native arm64 build for macOS as then the embedded python in Slicer will appropriately be an arm64 version that can use the arm64 macOS torch whls that provide the MPS support.

For now, macOS on Slicer must continue to use the intel x86_64 version that runs through the Rosetta 2 layer.

Once arm64 support is available we will post over on the following thread to provide an update:

3

Hello James,

The version of PyTorch that is running in Slicer3d has the MPS backend available. I noticed that the source code for MONAI Auto3dSeg does not include any reference to “mps” (lots of references to CUDA) so I think the code will also need to recognize “mps” as an optional device. Is my assumption correct?

Thank you!

Gene Risi

4

Yes if you have an Intel mac with amd GPU then torch can use the MPS. From my understanding, you have an Apple Silicon mac where the torch version I believe will not be able to use MPS with the graphics of Apple Silicon unless it is running an arm64 version (not an x86_64 version running through Rosetta 2).

5

If it would be helpful, I could test that, if it would just make more work for you, I will wait.

Gene