CUDA on MacOS 12.0.1 MacBook Pro M1 Chip

Hello dear all,

I was wondering if I could build CUDA from source even Mac doesn’t have an Intel GPU for the issue below:

conda install pytorch torchvision -c pytorch
# MacOS Binaries dont support CUDA, install from source if CUDA is needed

How can I recompile Pytorch from source to get gpu enabled?

Kind Regards,

I used CUDA 8.0 with the --override flag I believe.

I really wish nVidia would add a CUDA → OpenML wrapper for Apple’s tensor cores (branded as a “neural engine” but I’m not sure if that will actually happen xD

Guess I’ll change my device 😅

Thanks @Eirikr1848