如何在 Mac M1 MacBook Pro 上 运行 同时使用 TensorFlow 和 Torch?

How can I run both TensorFlow and Torch on Mac M1 MacBook Pro?

我在 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 对我来说至关重要,而 transformers 需要 TensorFlow 2.0 和 PyTorch,我需要回到我的旧电脑上进行编码。我很失望!

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

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

谢谢!

尝试从源代码构建,因为我能够在我的 MacBook pro 13 M1 上安装。

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

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

现在您可以使用新创建的虚拟环境打开一个 PyCharm 项目,并且您将能够导入 Tensorflow 和 Torch。然而,很多库会像 PyTorch 这样难以安装...

现在您可以使用 M1 pip 安装适用于 Mac 的 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”