macOS M2使用conda配置pytorch环境

一、conda

首先我们需要有conda，miniconda的安装配置流程可以看这个Mac M1/M2芯片安装Miniconda并搭建tensorflow2环境

二、进入官网

• PyTorch

选择对应的版本，复制下面的命令

conda install pytorch torchvision torchaudio -c pytorch

三、开始创建虚拟环境，下载pytorch

• 进入终端，输入

conda create -n torch-gpu python=3.9
conda activate torch-gpu
conda install pytorch torchvision torchaudio -c pytorch

第一条命令是创建名为torch-gpu，python版本为3.9的虚拟环境

第二条命令是激活环境

第三条命令就是安装pytorch

四、测试

python
import torch
torch.__version__
torch.device('mps')