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

Question

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

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

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

由于HuggingFace 转换器对我来说至关重要，而且转换器需要 TensorFlow 2.0 和 PyTorch，我需要回到我的旧计算机上进行编码。我非常失望！

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

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

谢谢！

标签： macostensorflowpycharmpytorchapple-m1

Answer 1

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

因此，如果有人尝试使用 PyCharm Apple Silicon 版本在 Mac M1 上同时运行 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 一样难以安装......