How to run on gpu in apple M1

Hi,

The GPU part of the M1 chip is not based on nvidia CUDA so it won’t work with our CUDA backend, but The CPU part should run just fine.

1 Like