目录

在 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 表示已正确安装。


上一篇:«

下一篇: »


评论

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注

*