Add support for 'mps' device for Apple Silicon

MIC-DKFZ / HD-BET

MRI brain extraction tool

Apache License 2.0

255 stars 61 forks source link

Open mrunibe opened 2 months ago

mrunibe commented 2 months ago

Pytorch now supports Conv3D on MacOS using MPS, allowing HD-BET to run with GPU support on Apple Silicon (M1/M2/M3).