在 Mac 上安装 PyTorch 夜间版本
1、更新 brew ,并安装依赖
首先需要更新 Mac 中的 brew
,以及安装依赖命令如下:
brew update
brew install cmake protobuf rust
2、安装 PyTorch 夜间版
根据个人的情况选择 PyTorch 夜间版选择以下其一命令进行安装:
仅 CPU 支持:
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
GPU支持(Apple Silicon / Metal 加速):
如果你的 Mac 是 M 芯片,使用该命令安装支持 Metal 加速的版本:
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
3、验证是否安装成功
以上命令安装完成后,可以写个 Python 脚本来执行,确认是否成功的安装了:
import torch
print(torch.__version__)
print(torch.backends.mps.is_available())
执行以上 Python 脚本,最后一行打印出 True
表示已正确安装。