Mac Pro M1测试PyTorch GPU

近年来，Apple推出的自家芯片M1引起了广泛关注。作为一款强大的处理器，M1芯片在性能和能效方面都有着显著的提升。在机器学习领域，PyTorch是一种常用的深度学习框架，而使用M1芯片进行PyTorch GPU加速成为了许多研究者和开发者的关注点。本文将详细介绍如何在Mac Pro M1上测试PyTorch GPU，并提供相应的源代码。

首先，我们需要确保在Mac Pro M1上正确安装了PyTorch和相关的依赖库。可以通过以下命令使用pip安装PyTorch：

pip install torch torchvision torchaudio

安装完成后，我们可以编写一个简单的Python脚本来测试PyTorch GPU的性能。以下是一个示例脚本：

import torch

# 检测是否支持GPU加速
if torch.cuda.is_available(