Macbook air m1使用gpu运行pytorch

下载对应版本的Pytorch

macbook m芯片使用的是Arm架构，目前m1芯片适配的加速后端是mps，其本事存在操作系统中，故只需下载对应版本的pytorch即可。

在pytorch官网上可以得到下载命令

你可以创建新的环境conda create -n new_env python=3.10
激活环境conda activate new_env

也可以使用自己原本的环境
查看当前环境conda env list
激活环境source activate 环境名

在mac终端输入pip3 install torch torchvision torchaudio

检查是否安装成功

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

可能的问题

出现问题&