MPS 后端

本文来自:
https://pytorch.org/docs/stable/notes/mps.html
https://pytorch.ac.cn/docs/stable/notes/mps.html


MPS 后端

mps 设备支持 在使用 Metal 编程框架的 MacOS 设备上,进行高性能 GPU 训练。

它引入了新的设备,将机器学习计算图和原语映射到 Metal Performance Shaders 图框架和 Metal Performance Shaders 框架提供的经过优化的内核上。

新的 MPS 后端扩展了 PyTorch 生态系统,并为现有脚本提供在 GPU 上设置和运行操作的功能。

要开始使用,只需将您的张量和模块移动到 mps 设备。

python 复制代码
# Check that MPS is available
if not torch.backends.mps.is_available():
    if not torch.backends.mps.is_built():
        print("MPS not available because the current PyTorch install was not "
              "built with MPS enabled.")
    else:
        print("MPS not available because the current MacOS version is not 12.3+ "
              "and/or you do not have an MPS-enabled device on this machine.")

else:
    mps_device = torch.device("mps")

    # Create a Tensor directly on the mps device
    x = torch.ones(5, device=mps_device)
    # Or
    x = torch.ones(5, device="mps")

    # Any operation happens on the GPU
    y = x * 2

    # Move your model to mps just like any other device
    model = YourFavoriteNet()
    model.to(mps_device)

    # Now every call runs on the GPU
    pred = model(x)

2024-07-16(二)

相关推荐
兮℡檬,1 小时前
房价预测|Pytorch
人工智能·pytorch·python
mCell6 小时前
从删库到跑路?这50个Linux命令能保你职业生涯
linux·windows·macos
贝塔西塔14 小时前
PytorchLightning最佳实践基础篇
pytorch·深度学习·lightning·编程框架
fhf14 小时前
2025年了你会卸载Macbook上的应用吗?
macos·shell
小猪和纸箱14 小时前
通过Python交互式控制台理解Conv1d的输入输出
pytorch
归辞...15 小时前
「iOS」——内存五大分区
macos·ios·cocoa
穆雄雄18 小时前
备份一下我的 mac mini 的环境变量配置情况
macos
墨染枫21 小时前
pytorch学习笔记-使用DataLoader加载固有Datasets(CIFAR10),使用tensorboard进行可视化
pytorch·笔记·学习
秃然想通1 天前
mac电脑搭载c、c++环境(基于vs code)
macos
归辞...1 天前
「iOS」——GCD其他方法详解
macos·ios·cocoa