苹果mac m1,m2芯片安装 pytorch和tensorflow的GPU版本

一、下载M芯片的anaconda，并安装

二 、安装GPU版本的pytorch

1.安装 Xcode

xcode-select --install

2.创建环境

conda create -n torch-gpu python=3.9

conda activate torch-gpu

3.打开pytorch官网复制命令, 注意:在mac m上，device是’mps’ 而不是’cuda’, mac的MPS支持MacOS 12.3+

 4.测试

import torch
import math

print(torch.backends.mps.is_available())
# True
print(torch.backends.mps.is_built())
# True

三、安装GPU版本的tensorflow