用 conda 创建一个 arm64 的 Python 环境在这个 conda 环境里用 pip 安装 PyTorch用 MPS 验证是否启用了 Apple GPU这是因为 PyTorch 官方当前在 macOS 上推荐的包管理方式是 pip并注明最新稳定版要求 Python 3.10macOS 安装页也直接给出了 pip3 install torch torchvision。直接执行下面这套命令就行。1) 创建新环境推荐 Python 3.11conda create -n torch-m2 python3.11 -y2) 激活环境conda activate torch-m23) 确认当前 Python 是 arm64python -c “import platform; print(platform.platform()); print(platform.machine())”4) 安装 PyTorchpip install torch torchvision torchaudio5) 验证安装和 MPSpython - ‘PY’import torchprint(“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可以这样写importtorch devicempsiftorch.backends.mps.is_available()elsecpuprint(use device:,device)