Skip to content

Instantly share code, notes, and snippets.

@pipme

pipme/Jax installation on Mac M1.md

Created January 19, 2022 14:12
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save pipme/dc086aa3836488502283c0cd030317fb to your computer and use it in GitHub Desktop.
Save pipme/dc086aa3836488502283c0cd030317fb to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Jax installation on Mac M1.md

Install Miniforge3-MacOSX-arm64 first, then

conda create --name py39_forge python=3.9
conda activate py39_forge
conda install numpy scipy
pip install -U https://files.pythonhosted.org/packages/a1/70/74886d4edc355a375532f6cfe18a7f86a2db5093b374270fbd203414e404/jaxlib-0.1.75-cp39-none-macosx_11_0_arm64.whl
pip install jax
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment