Support for Apple GPUs (Apple M2 Max)

MIC-DKFZ / HD-BET

MRI brain extraction tool

Apache License 2.0

283 stars 65 forks source link

Open ncbss opened 1 week ago

ncbss commented 1 week ago

Hi there, Thanks for this awesome software.

I saw a couple of comments about making HD-BET compatible with Apple GPUs but it was not clear whether this is the case. Can you guys confirm?

Thanks

mrunibe commented 1 week ago

I have successfully used HD-BET on Apple Silicon using MPS backend (https://github.com/MIC-DKFZ/HD-BET/issues/54). The performance is good, much faster than running on CPU.

However, some minor modifications are required, see corresponding PR:

With this change you can then run it with -device ms

@FabianIsensee any plans to merge this PR?