search

alembics / disco-diffusion

Other
7.48k stars 1.13k forks source link

Does the mac work? Is the GPU that the M1 processor carries okay? #84

Closed 252552899 closed 2 years ago

252552899 commented 2 years ago

Does the mac work? Is the GPU that the M1 processor carries okay?

aletts commented 2 years ago

Currently, it wouldn't use the M1's GPU. So in order for it to work locally (rather than on Colab as most people do) on a Mac at all, you'd have to limit it to using the CPU by checking useCPU. When it's just using the CPU, it's incredibly slow.

There was recently an announcement from PyTorch that the next version of PyTorch will support the M1's GPU. So we'll see what happens when that comes. Until now, Nvidia's CUDA is basically what everyone uses for GPU acceleration.