Skip to content

Pytorch on Apple Silicon GPU

Felix Soubelet requested to merge pytorch-metal into master

This adds a block on the Apple Silicon page of the website with instructions on how to get PyTorch to use the Apple Silicon GPU through Metal.

This is very recent news (today!) but I've included simple examples of how to tell PyTorch to use the device, which was not necessary with Tensorflow.

Let me know if that looks good.

Merge request reports