m1配置pytorch gpu环境
时间: 2024-01-07 18:05:44 浏览: 172
如何搭建pytorch环境
要在 M1 Mac 上配置 PyTorch 的 GPU 环境,您需要执行以下步骤:
1. 安装 Homebrew:打开终端并运行以下命令:
```bash
/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
```
2. 安装 Miniforge:在终端中执行以下命令安装 Miniforge,它是一个轻量级的 Anaconda 版本,适用于 M1 架构:
```bash
brew install miniforge
```
3. 创建新的 conda 环境:在终端中执行以下命令创建一个新的 conda 环境并激活它:
```bash
conda create -n pytorch python=3.8
conda activate pytorch
```
4. 安装 PyTorch 和 torchvision:在 conda 环境中执行以下命令安装 PyTorch 和 torchvision:
```bash
conda install pytorch torchvision -c apple -c pytorch -c conda-forge
```
现在,您已经成功地在 M1 Mac 上配置了 PyTorch 的 GPU 环境。您可以在 conda 环境中编写和运行 PyTorch 代码。
阅读全文