Sudhanva Narayana
Sudhanva Narayana

Sudhanva Narayana

Install PyTorch on Apple M2 (M1, Pro, Max) with GPU

Install PyTorch on Apple M2 (M1, Pro, Max) with GPU

Sudhanva Narayana's photo
Sudhanva Narayana
·May 23, 2022·

1 min read

Subscribe to my newsletter and never miss my upcoming articles

Play this article

Table of contents

  • Note:
  • Installing Miniforge
  • Installing PyTorch
  • Verifying Installation
  • Credits

Note:

  • Uninstall Anaconda/Anaconda Navigator and other related previously installed version of conda-based installations.
  • Anaconda and Miniforge cannot co-exist together.
  • However, if you have previous installations of PyTorch with Miniforge, you can continue to use that without uninstalling it.
  • We will install the GPU version in a new conda environment anyway.
  • Keep in mind that the GPU version is still a nightly build and expect it to have breaking changes for the next couple of months.

Installing Miniforge

  • Install miniforge from brew: brew install miniforge
  • Create an anaconda environment: conda create -n pt122
  • Activate the environment: conda activate pt122
  • Install Python: conda install python

Installing PyTorch

  • Run: pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu to install PyTorch (Nightly) dependencies
  • Execute: pip install pandas jupyterlab to install relevant dependencies to run sample code.
  • Run: conda activate pt122

Verifying Installation

  • Execute: jupyter-lab to open a Jupyter Notebook and run the following code:

    import torch torch.device("mps") torch.version torch.tensor([1,2,3], device="mps")

Credits

 
Share this