M1 安装pytorch
时间: 2023-08-26 11:17:09 浏览: 57
您好!感谢您的提问。要在 M1 芯片的 Mac 上安装 PyTorch,您可以按照以下步骤进行操作:
1. 确保您的 Mac 已经安装了 Homebrew。如果尚未安装,请打开终端并运行以下命令来安装 Homebrew:
`/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"`
2. 确保您的 Mac 上已经安装了 Xcode 的最新版本。您可以在 App Store 中下载并安装 Xcode。
3. 打开终端,并运行以下命令来安装 Miniforge(Miniconda 的一个变体):
`arch -x86_64 /bin/bash -c "$(curl -fsSL https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-MacOSX-x86_64.sh)"`
4. 安装 Rosetta 2,以便在 M1 芯片上运行 x86_64 架构的软件。在终端中运行以下命令:
`softwareupdate --install-rosetta`
5. 创建一个新的 conda 环境并激活它。在终端中运行以下命令:
```shell
conda create -n pytorch_m1 python=3.8
conda activate pytorch_m1
```
6. 使用 conda 安装 PyTorch 和 torchvision。在终端中运行以下命令:
```shell
conda install pytorch torchvision torchaudio -c pytorch -c apple
```
这样,您就可以在 M1 芯片的 Mac 上成功安装 PyTorch 了。希望对您有所帮助!如有任何问题,请随时向我提问。