Mac GPU MPS常用方法

本文主要是介绍Mac GPU MPS常用方法，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Requirements

Mac computers with Apple silicon or AMD GPUs
macOS 12.3 or later
Python 3.7 or later
Xcode command-line tools: xcode-select --install

判断是否可用

import torch
if torch.backends.mps.is_available():mps_device = torch.device("mps")x = torch.ones(1, device=mps_device)print (x)
else:print ("MPS device not found.")

这篇关于Mac GPU MPS常用方法的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---