Feature: add option for selecting compute device

royerlab / napari-segment-anything

Segment Anything Model (SAM) native Qt UI

Apache License 2.0

190 stars 17 forks source link

Open kephale opened 11 months ago

kephale commented 11 months ago

I ran into this for reasons related to the MPS device issues (I'm on a MBP M1) https://github.com/royerlab/napari-segment-anything/issues/10.

It would be nice to be able to specify the compute device, particularly to make it easier to test on CPU.

JoOkuma commented 11 months ago

Hey @kephale, we can add a combo box to the widget to allow that. I don't know when I'll have time to do this, though.

kephale commented 11 months ago

No rush @JoOkuma. I just wanted to track the feature.