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

conda create -n torch-gpu python=3.11
conda activate torch-gpu

import torch
import math

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

xcode-select --install

conda create -n tensorflow-gpu python=3.11

conda activate tensorflow-gpu

conda install -c apple tensorflow-deps

python -m pip install tensorflow-macos==2.9

python -m pip install tensorflow-metal==0.6

import tensorflow as tf
print(tf.test.is_gpu_available())