python-3.x - python3 在 Apple silicon M1 上安装 Tensorflow

Question

我在 Apple Silicon M1 上安装了 macOS Big Sur，但我无法在 python3 中安装 Tensorflow。我删除了 xcode python3 并安装了 brew arm64 python3(x86 python3 也不起作用)

我检查成功的64位bis版本

python3 -c "import sys; print(sys.version)" or python -c "import struct; print(struct.calcsize('P')*8)"

3.8.7 (default, Dec 30 2020, 02:09:32)[Clang 12.0.0 (clang-1200.0.32.28)]

无论如何这都能工作吗，或者谁知道如何工作？

Answer 1

Apple M1是ARM64架构的处理器，而TensorFlow的所有pip包都是为x86_64架构编译的。 (不包括 raspberry pi 包，但它们无论如何都不会与 MacO 兼容)。

如果我们在 install page 上查看 pip 包的名称在 TensorFlow 中，大多数包含 x86_64 或 amd64，这表明它们是针对 x86_64 架构构建的。

如果您想在 M1 上运行 TensorFlow，您需要:

• 通过 Rosetta 2 从针对 x86_64 的源代码编译 TensorFlow。不幸的是，根据 this issue on github , Rosetta 2 不支持在 TensorFlow 的 pip 构建中启用的 AVX 指令集，因此需要从源代码重建。
• 在 Apple 开发的 M1 上使用对 TensorFlow 的实验性支持，您可以在 this github repository 上找到它.请注意，如果采用这种方式，某些 python 包可能与 M1 不兼容/不可用。

请注意，截至 2021/01/04，the Apple M1 is not a supported architecture by the TensorFlow team :

We currently cannot support Mac ARM. There is less than one developer than can focus on build issues on all 3 operating systems.

Hence, support has to come from the the community. There is SIG Build that focuses on build related issues on several other platforms, leaving us to only focus on the pip packages we officially bless.