Open pmaojo opened 2 years ago
I'm not sure if it can work with a Macbook Pro M1.
There are a few issues in my mind that would need to be overcome:
tensorflow
image, however.tensorflow-gpu
image uses CUDA. I'm not sure if there's an equivalent for the M1.Let me know if you investigate this in more detail. It would be great to have a working arm64 version.
According to this it does run, however, it is kind of slow and I ran into some issues that I didn't bother resolving. Now I just run it on an EC2 instance.
I was already afraid this would not work fine on a Arm based Mac. It does work, but the fact that it's running as an amd64 image (emulation) and no gpu available makes the performance really bad. I did generate an image on my Macbook Pro M1. It was at least a hour for generating one image.
I am having this error on build: