How to use GPU on Mac M1/2

Install xcode tools

xcode-select --install

Install llvm

brew install llvm libomp

Install PyTorch for M1/2

pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu