如何在 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”