Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support for Apple GPUs (Apple M2 Max) #61

Open
ncbss opened this issue Nov 9, 2024 · 2 comments
Open

Support for Apple GPUs (Apple M2 Max) #61

ncbss opened this issue Nov 9, 2024 · 2 comments

Comments

@ncbss
Copy link

ncbss commented Nov 9, 2024

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
Copy link

mrunibe commented Nov 13, 2024

I have successfully used HD-BET on Apple Silicon using MPS backend (#54). The performance is good, much faster than running on CPU.

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

#55

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

@FabianIsensee any plans to merge this PR?

@FabianIsensee
Copy link
Member

We had a new release on Friday which now bases hd-bet on nnU-Net. With that we should have support for mps. I could not test this though as I don't have access to a mac

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants