Skip to content

Instantly share code, notes, and snippets.

@dkirkby

dkirkby/gist:a3f7a6caa7495c5a508f1c33206b87d1

Last active July 3, 2024 12:14
Show Gist options
  • Save dkirkby/a3f7a6caa7495c5a508f1c33206b87d1 to your computer and use it in GitHub Desktop.
Save dkirkby/a3f7a6caa7495c5a508f1c33206b87d1 to your computer and use it in GitHub Desktop.
Download ZIP
Apple Silicon ML
Raw
gistfile1.txt
# https://developer.apple.com/metal/jax/
conda create -n apple-ml python=3.10 pip ipython jupyter jupyterlab ipykernel numpy wheel scipy pandas matplotlib
conda activate apple-ml
python -m pip install ml-dtypes==0.2.0 jax-metal
# Successfully installed jax-0.4.11 jax-metal-0.0.4 jaxlib-0.4.11 ml-dtypes-0.2.0 opt-einsum-3.3.0
## This installs a newer version of jaxlib that is incompatible with jax-metal ??
##python -m pip install -q -U flax
# https://developer.apple.com/metal/pytorch/
conda install pytorch torchvision torchaudio -c pytorch-nightly
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment