M2安装pytorch

M2芯片是苹果公司开发的一种专为Mac电脑设计的芯片，它具有更高的性能和更低的功耗。如果你想在M2芯片的Mac电脑上安装pytorch，你可以按照以下步骤进行操作：

1. 下载并安装Anaconda：在官方网站上下载M芯片的Anaconda安装包，并按照指南进行安装。

2. 创建环境：使用conda命令创建一个新的环境，例如"torch-gpu"，并激活该环境。

3. 安装GPU版本的pytorch：在终端中输入官方提供的安装命令，并注意在M芯片上，设备应该是"mps"而不是"cuda"。例如：`conda install pytorch torchvision torchaudio -c pytorch -c apple/label/macosarm64 -c conda-forge`

4. 进行测试：导入torch模块和math模块，并打印出torch.backends.mps.is_available()和torch.backends.mps.is_built()的结果，确保安装成功。