python 验证macbook的 mps gpu是否存在

直接在 Mac 上使用 TensorFlow 加速机器学习模型的训练。安装基础 TensorFlow 和 tensorflow-metal PluggableDevice 以在 Mac GPU 上使用 Metal 加速训练

import torch

if torch.backends.mps.is_available():
    print("M1 GPU is available")
else:
    print("M1 GPU is not available")

import torch 
print(torch.backends.mps.is_available()) 
print(torch.backends.mps.is_built())