在Apple Silicon的GPU上运行PyTorch

根据PyTorch文档信息，PyTorch v1.12开始支持在苹果芯片上进行GPU加速的模型训练。安装时有两个要求：

• MacOS版本在12.3以上。本人使用的版本为12.5，芯片为Apple M1 Pro。
• PyTorch要求安装Preview(Nightly)版本。

PyTorch安装脚本：

conda install pytorch torchvision torchaudio -c pytorch-nightly

安装完成后，在Terminal上进入python环境，并执行以下脚本确认是否可以使用GPU加速。

Python 3.9.12 (main, Jun  1 2022, 06:36:29) 
[Clang 12.0.0 ] :: Anaconda, Inc. on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.backends.mps.is_available()
True
>>> 

如果torch.backends.mps.is_available()返回True，则表示可以进行GPU加速。

待补充。。。