If you have an Apple M1 or M2 and don’t take advantage of its GPU, you may be missing out! Twitter user Santiago has written instructions to allow TensorFlow to use the GPU on M1 and M3-based Apple computers.
The simplified steps:
- Install Homebrew
- Download Miniforge3
- Install and activate Miniforge3.
- Restart your terminal window.
- Create and activate a Conda environment
- Install TensorFlow, Jupyter, and any libraries you need
- Run Jupyter Notebook
- Run the code above in one of the cells of the notebook
The GPU should be listed as part of the devices available.