search

volume-em / empanada-napari

Panoptic segmentation algorithms for 2D and 3D electron microscopy in napari
https://empanada.readthedocs.io/en/latest/empanada-napari.html
BSD 3-Clause "New" or "Revised" License
20 stars 8 forks source link

Support pytorch acceleration on M1 Mac hardware #17

Open GenevieveBuckley opened 2 years ago

GenevieveBuckley commented 2 years ago

DO NOT MERGE

  1. We're waiting for the next official release of pytorch including M1 Mac hardware accleration (right now I'm using the nightly build to test things)
  2. The MitoNet model is not supported on the M1 Mac mps backend, so things are broken now because of that as well

The pytorch nightly (I'm using version '1.13.0.dev20220616') now supports acceleration on M1 Mac hardware. This branch experiments with adding support for this in empanada-napari.

Cross reference: https://github.com/volume-em/empanada/pull/14