如何在 Mac M1 MacBook Pro 上同时运行 TensorFlow 和 Torch？

Question

我在 MacBook Pro M1 上遇到了一些问题。我认为开始对其进行编码会更容易，显然它还不是 ML 野兽...

我需要在 Python 上同时使用 PyTorch 和 TensorFlow。我已经安装了TensorFlow 2.0 for Mac OS。

问题是：当您使用 x86_64 终端时，TensorFlow 将无法工作。 （所以它不适用于 PyCharm）。但是，我可以从 arm 终端导入 TensorFlow 2.0。 矛盾的是，PyTorch 不会安装在 arm 终端上，只能安装在 x86_64 终端上。因此，在同一个 Python 终端上，我无法同时导入 torch 和 TensorFlow 2.0。

由于HuggingFace transformers 对我来说至关重要，而且变形金刚需要 TensorFlow 2.0 和 PyTorch，我需要回到我的旧电脑上进行编码。我很失望！

有人在 Mac M1 设备上成功导入了 PyTorch 和 TensorFlow？

有谁知道是否有办法强制 PyCharm 使用 arm 终端，以便我可以在我的 M1 MPB 上使用 TensorFlow 2.0 on PyCharm？

谢谢！

Answer 1

经过一番研究，我找到了这个答案：https://github.com/pytorch/pytorch/issues/48145。

因此，如果有人尝试在 Mac M1 上使用 PyCharm Apple Silicon 版本同时运行 Tensorflow 和 Torch，请按照以下步骤进行：

• 使用 Tensorflow Mac OS 创建新的虚拟环境
• 来自 终端（不是来自 PyCharm，我遇到了一个错误），使用 sudo 权限，安装用于 Torch 的 whl 文件，来自 GitHub 问题：https://github.com/wizyoung/AppleSiliconSelfBuilds/blob/main/builds/torch-1.8.0a0-cp39-cp39-macosx_11_0_arm64.whl

现在您可以使用新创建的虚拟环境打开一个 PyCharm 项目，并且可以同时导入 Tensorflow 和 Torch。但是，很多库像 PyTorch 一样难以安装......

Answer 2

尝试从源代码构建，因为我可以安装在我的 MacBook pro 13 M1 上。 PyTorch installation from source