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

Question

我在 MacBook Pro M1 上遇到了一些问题。我认为开始编码会更容易，显然它还不是机器学习野兽......

我需要在 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 对我来说至关重要，而且 Transformers 需要 TensorFlow 2.0 和 PyTorch，所以我需要回到我的旧计算机上进行编码。我很失望！

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

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

谢谢！

Answer 1

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

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

• 使用 Tensorflow Mac OS 创建新的虚拟环境
• 从 终端（不是来自 PyCharm，我遇到了错误），使用 sudo 权限，从 GitHub 问题安装用于 Torch 的 whl 文件：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