trying to fix M1 apple #523

MouseLand · Jan 12, 2023 · ffacce8 · ffacce8
1 parent 9a433ae
commit ffacce8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cellpose/core.py b/cellpose/core.py
@@ -79,6 +79,7 @@ def assign_device(use_torch=True, gpu=False, device=0):
         try:
             device = torch.device('mps')
             gpu=True
+            cpu=False
             core_logger.info('>>>> using GPU')
         except:
             cpu = True