Merge pull request #1466 from MouseLand/mps_docs

fix docs for mps #1455

Author: carsen-stringer

cellpose/cli.py

@@ -26,10 +26,10 @@ def get_arg_parser():
     # settings for CPU vs GPU
     hardware_args = parser.add_argument_group("Hardware Arguments")
     hardware_args.add_argument("--use_gpu", action="store_true",
-                               help="use gpu if torch with cuda installed")
+                               help="use gpu or mps if torch with cuda installed")
     hardware_args.add_argument(
         "--gpu_device", required=False, default="0", type=str,
-        help="which gpu device to use, use an integer for torch, or mps for M1")
+        help="which gpu device to use in pytorch, specified as an integer")
 
     # settings for locating and formatting images
     input_img_args = parser.add_argument_group("Input Image Arguments")

docs/installation.rst

@@ -31,13 +31,11 @@ this code at the beginning of your notebook before you import cellpose:
 M1-M3 Mac installation
 ~~~~~~~~~~~~~~~~~~~~~~~
 
-Support for M1-M3 should work out-of-the-box with Cellpose now! Please submit an issue if it's not working.
-
-From the command line you can choose the Mac device explicitly with
+If you specify ``--use_gpu`` on the command line, Cellpose will automatically use the MPS acceleration for Apple Silicon if you have a compatible version of PyTorch installed. Please submit an issue if it's not working.
 
 ::
 
-   python -m cellpose --dir path --gpu_device mps --use_gpu
+   python -m cellpose --dir path --use_gpu
 
 AMD GPU ROCm installation
 ~~~~~~~~~~~~~~~~~~~~~~~~~~