PyTorch使用Apple M1芯片加速训练
摘要
安装 Miniconda3
注意：要启动苹果 M1 系列芯片加速需安装 arm64 架构原生 python 环境！
curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh
sh Miniconda3-latest-MacOSX-arm64.sh
【conda 环境配置这里略过。。。】
创建并激活一个新的 conda 环境
conda create -n torch python=3
conda activate torch
安装 nightly 版本 PyTorch
pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
验证 Apple mps 后端已支持
import torch
torch.backends.mps.is_available()
quit()
后续
PyTorch 现在支持了 Apple M1 系列芯片加速，从简单的测试来看已经有相当不错的表现（虽然综合来看目前还赶不上 1080ti，但是差距已经小于一个数量级），未来可期！
