Setup LiveBook/Elixir to run Nx lib with Torchx (with GPU) on Macbook(Apple Silicon)

If you want to run Elixir Nx library for machine learning on Macbook/Mac Mini/Mac Studio with Metal (GPU) for Apple Silicon you can simple follow steps.

1. Setup and config Pytorch.

Install asdf if you don't have one.

Follow below script to install Python3 by asdf

asdf plugin-add python
asdf install python 3.x.x
asdf global python 3.x.x

pip3 install torch torchvision

Verify with simple Python script:

Run Python:

python

add text in Python:

import torch
x = torch.rand(5, 3)
print(x)

You …

apple apple silicon config elixir gpu install library mac macbook machine machine learning machinelearning metal plugin python3 pytorch setup silicon simple studio