Currently PyTorch can not be compiled natively on Apple Silicon, because it is reported as “arm64” architecture and many third-party libraries only support ARMv8 or aarch64
1 thought on “Enable PyTorch compilation on Apple Silicon”
Comments are closed.
I got an m1 Macbook Air and a 2018 Intel Macbook Air right here. For an identical CNN for MNIST, the Intel Macbook took some 1975 seconds whereas the m1 took 2100 seconds.
So the Apple Silicon Macbook is slower, but on paper, it does ‘run smoothly’ I suppose.
PS: I’m a complete noob to developing on PyTorch but I’d be glad to do some help here or there since I have one of these Macs if the team can just help point me on what to do.