Local PyTorch/Tensorflow Development on M1 MacBooks

Conda environments for local PyTorch and/or Tensorflow development on Apple Silicon Mac Books.

Prequisites:

• An Apple Silicon Mac Book
• miniforge (and conda) for M1:
  • brew install miniforge
  • conda init zsh

To import, run:

conda env create -f pytorch_mac_m1.yml

and/or

conda env create -f tensorflow_mac_m1.yml

To list environments, type:

conda env list

To activate, run:

conda activate pytorch

or

conda activate tf2

NOTE: Do not update the PyTorch environment using conda env update -n pytorch --all , as long as the Apple Silicon support is available only in nightly builds. Such an update would override the pytorch and torchvision with the versions from the stable channel. And that one, at the moment of writing, doesn't support the Apple Silicon accelerated training natively.