软件配置 | mac M1安装pytorch踩坑不完全记录

本文记录在 macOS 11.4 上安装 pytorch 的过程。

更新：2023 / 3 / 4

---

---

pytorch

因为 brew 和 miniforge3 我已经提前安装过，所以这里不再记录了。没有安装过的同志也可以参考这里。

安装

• 基于python3.8创建一个名为pytorch_env的新虚拟环境
  conda create --name pytorch_env python=3.8
• 激活上面创建的虚拟环境
  conda activate pytorch_env

• 在anaconda官网 找到 pytorch, 尝试 conda install pytorch torchvision -c pytorch，会如下所示的错误 …

• 在 pytorch 官网看到可以通过 pip 来安装 pytorch，pip3 install torch torchvision

验证

根据这里，验证是否 pytorch 已经安装成功。
验证结果呢，如下所示：

从结果看是 CUDA 存在问题。

参考这里，可以查到

但其实如果是 m1 芯片的话，在 pytorch 的官方论坛上有一个帖子 里工程师说：

并且在 pytorch 的 github 主页可以看到，暂时还不支持在 m1 gpu 上跑 pytorch …

所以先看看能不能跑 cpu 吧。
结果如下所示：

更新！

参考这篇这里回答可知，可以在 Apple Silicon M1 GPU 的 Pytorch 啦。
但是对环境有要求，要求 macOS 12.3 以上和 Python 3.7 以上。

安装完成后，在 Terminal 上进入 python 环境，通过以下方式确认 python 是否为 arm64 架构 ¹：

~ % python
Python 3.8.11 (default, Jul 29 2021, 14:57:32) 
[Clang 12.0.0 ] :: Anaconda, Inc. on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import platform
>>> print(platform.uname()[4])
arm64

---

参考链接

M1 mac安装PyTorch的完整步骤指南
Mac怎么查看Homebrew是否安装
conda activate激活环境出错的解决办法
GET STARTED
Apple MacBook M1 芯片 Anaconda安装 Tensorflow Pytorch
（Ubuntu/Window）pytorch验证（pytorch是否安装成功）
Pytorch错误：Torch not compiled with CUDA enabled
How to run on gpu in apple M1
GPU acceleration for Apple’s M1 chip?
苹果 M1 芯片的神经单元可否用于训练 Pytorch 深度学习网络模型？

---

1. 在Apple Silicon的GPU上运行PyTorch ↩︎