如何在 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，我需要 go 回到我的旧计算机上进行编码。 我非常失望！

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

And does anyone know if there is a way to force PyCharm to use an arm terminal, so I can use TensorFlow 2.0 on PyCharm on my M1 MPB?

谢谢！

Answer 1

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

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

• 使用 Tensorflow Mac OS 创建一个新的虚拟环境
• From terminal (not from PyCharm, I got an error), with sudo rights, install the whl file for torch, from the GitHub issue: 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 从源码安装

Answer 3

现在您可以为带有 M1 的 Mac 安装 pip PyTorch！

https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/

你基本上需要使用这个

pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu

并在您的代码中将“cuda”替换为“mps”