Flux.jl GPU support for M1 Mac?

1k Views Asked by At

I am currently using a laptop with the M1 Mac chip. The Flux.jl docs only mention support for Nvidia GPU's. Is it possible to train my models using my local GPU or is there no support for the M1 Mac yet? Note: Julia 1.7 beta 3 is currently being shipped with M1 Mac support so it seems like this would be feasible but I am not a GPU expert.

1

There are 1 best solutions below

2
On

I'm not sure that there is much reason (so far) to support an 8 core GPU on the M1 when typical Nvidia CUDA core counts are in the hundreds or the thousands. That may change with future CPU's, perhaps.