mac M芯片安装pytorch

用 conda 创建一个 arm64 的 Python 环境

在这个 conda 环境里用 pip 安装 PyTorch

用 MPS 验证是否启用了 Apple GPU

这是因为 PyTorch 官方当前在 macOS 上推荐的包管理方式是 pip，并注明最新稳定版要求 Python 3.10+；macOS 安装页也直接给出了 pip3 install torch torchvision。

直接执行下面这套命令就行。

1) 创建新环境（推荐 Python 3.11）

conda create -n torch-m2 python=3.11 -y

2) 激活环境

conda activate torch-m2

3) 确认当前 Python 是 arm64

python -c “import platform; print(platform.platform()); print(platform.machine())”

4) 安装 PyTorch

pip install torch torchvision torchaudio

5) 验证安装和 MPS

python - <<‘PY’
import torch
print(“torch version:”, torch.version)
print(“mps built:”, torch.backends.mps.is_built())
print(“mps available:”, torch.backends.mps.is_available())

if torch.backends.mps.is_available():
x = torch.ones(1, device=“mps”)
print(“MPS OK:”, x)
else:
print(“MPS 不可用，请检查 macOS 版本 / Python 架构 / 安装方式”)
PY

如果你在代码里想优先使用 M2 的 GPU，可以这样写：

import torch

device = "mps" if torch.backends.mps.is_available() else "cpu"
print("use device:", device)