判断 Mac显卡 mps 是否可以正常运行

这篇具有很好参考价值的文章主要介绍了判断 Mac显卡 mps 是否可以正常运行。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。


判断 Mac显卡 mps 是否可以正常运行,Bug特攻队,Mac显卡,原力计划

Mac MPS介绍

使用 Apple 的 Metal Performance Shaders (MPS) 作为 PyTorch 的后端来启用加速 GPU 训练。 MPS 后端扩展了 PyTorch 框架,提供了在 Mac 上设置和运行操作的脚本和功能。 MPS 使用针对每个 Metal GPU 系列的独特特性进行微调的内核来优化计算性能。 新设备将机器学习计算图和基元映射到 MPS 提供的 MPS Graph 框架和优化内核上。

随着PyTorch v1.12版本的发布,开发人员和研究人员可以利用 Apple 的GPU进行明显更快的模型训练,在部分模型的训练上往往比Mac 的 CPU有着数倍的提升。

  • (Mac M1芯片为了追求高性能和节能,在底层设计上使用的是一种叫做arm架构的精简指令集,不同于Intel等常用CPU芯片采用的x86架构完整指令集。所以有些基于x86指令集开发的软件不能直接在Mac M1芯片电脑上使用。)
  • 需要注意的是,使用Mac M1芯片加速 pytorch 不需要安装 cuda后端,因为cuda是适配nvidia的GPU的,Mac M1芯片中的GPU适配的加速后端是mps,在Mac对应操作系统中已经具备,无需单独安装。只需要安装适配的pytorch即可。

判断 Mac mps 是否可以正常运行

直接在 shell 窗口执行代码,输出为 True 则说明 可以正常运行

python -c "import torch;print(torch.backends.mps.is_built())"

或者写在一个代码中执行:

import torch
import math

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

Mac 指定 mps 进行推理示例

在Mac M1下的GPU称作mps,它类似于Nvidia的cuda。如果你想在Mac M1下使用GPU进行深度学习的训练,只需要将运算指定到mps上运行即可。

import torch
model = torch.nn.Linear(5,1).to('mps')
x = torch.randn(5).to('mps')
y = model(x)
print(y.shape)
  • 如上代码所示,该代码首先定义了一个最简单5x1的线性模形,并告诉Pytorch它在Mac M1的GPU上运行;
  • 然后给该模形创建一个输入,该输入也是在GPU上,是一个5x5的tensor,最后输出运算结果的维度。
  • 如果上面的代码可以正常运行,说明你Mac上的GPU确实可以正常工作。
  • 本章节参考博文: https://avdancedu.com/eaff035c/

Mac M1 GPU 训练 相比 CPU 速度提升对比

Apple M1 Ultra, 20-core CPU, 64-core GPU 128GB of RAM, and 2TB SSD. Tested with macOS Monterey 12.3, prerelease PyTorch 1.12, ResNet50 (batch size=128), HuggingFace BERT (batch size=64), and VGG16 (batch size=64).

  • 详情参考链接

判断 Mac显卡 mps 是否可以正常运行,Bug特攻队,Mac显卡,原力计划


📙 开心搬砖每一天


  • 🎉 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️
  • ❤️ 过去的每一天、想必你也都有努力、祝你披荆斩棘、未来可期
  • 🍊 深度学习模型训练推理——基础环境搭建推荐博文查阅顺序【基础安装—认真帮大家整理了】
  • 🍊 计算机视觉:硕博士,交流、敬请查阅
  • 🍊 点赞 👍 收藏 ⭐留言 📝 都是博主坚持写作、更新高质量博文的最大动力!

判断 Mac显卡 mps 是否可以正常运行,Bug特攻队,Mac显卡,原力计划文章来源地址https://www.toymoban.com/news/detail-558127.html

到了这里,关于判断 Mac显卡 mps 是否可以正常运行的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 【线性代数】如何判断矩阵是否可以相似对角化

    第一步,看是不是实对称矩阵,如果是实对称矩阵,立即推可相似对角化,如果不是实对称矩阵,看第二步; 第二步,求方阵的n个特征值,如果特征值彼此都不相同,也就是都是单根的话,立即推可相似对角化,如果有重根,看第三步; 第三步,来验证k重根是不是具备k个

    2024年02月11日
    浏览(50)
  • 哪些方法可以判断出一个有向图是否有环

    使用 深度优先遍历 ,若从有向图上的某个顶点u出发,在 DFS(u)结束之前出现一条从顶点v到u的边,由于v在生成树上是u的子孙,则图中必定存在包含u和v的环,因此深度优先遍历可以检测一个有向图是否有环。 拓扑排序 时,当某顶点不为任何边的头时才能加入序列,存在环时环中的

    2024年02月12日
    浏览(46)
  • Python中可以用三种方法判断文件是否存在

    前言 大家早好、午好、晚好吖 ❤ ~欢迎光临本文章 通常在读写文件之前,需要判断文件或目录是否存在,不然某些处理方法可能会使程序出错。 所以最好在做任何操作之前,先判断文件是否存在。 这里将介绍三种判断文件或文件夹是否存在的方法,分别使用os模块、Try语句

    2024年02月02日
    浏览(48)
  • Mac机RVM安装,手动下载安装,经过验证可以正常使用

    1、正常方法(不容易成功),我自己就卡了两周(因为墙的问题一直搞不定) 中国境内访问 虽然可以访问,但是下载使用会被强,可能有一些翻越的方法,但是不容易搞 (1)下载资源,我博客里找吧,正在审核 (2)、把资源存放到一个位置上 例:我存放的位置是 (3)、

    2024年01月19日
    浏览(49)
  • vue3在运行时路由可以正常跳转,但是页面空白,刷新页面恢复正常[已解决]

    vue3在运行时路由可以正常跳转,但是页面空白,刷新页面恢复正常,在网上寻找很久说是keepAlive的原因,后来看了看我的代码~ 原来的代码: 改后的代码: 由此可见,根元素一定要由一个div去包裹住! template必须有且只能有一个div 在这样才会正常显示~ 原理: 这里我们要先

    2023年04月27日
    浏览(59)
  • Linux shell:脚本判断进程是否在运行

    一.命令说明 ps aux | grep 进程名 | grep -v grep :进程存在则输出信息,不存在则没输出 ps -ef | grep 进程名 | grep -v grep | wc -l :进程存在则输出行数(也就是数量),不存在则输出0 二.运行示例demo 方法一: 新建脚本:touch test.sh 添加如下内容: 赋予权限:chmod 777 test.sh 运行结果

    2024年02月11日
    浏览(59)
  • Tomcat报404问题解决方案大全(包括tomcat可以正常运行但是报404)

    文章概览: 本文将解决一下问题: 问题有先后顺序 Tomcat未配置环境变量 IIs访问权限问题 端口占用问题 文件缺少问题(tomcat可以正常运行但是报错404) 正确的运行框: 正确的网页: 首先你要报个404玩玩; 这里怎么解决呢,将分为一下几类: 根据这个问题可以参考作者之

    2024年02月03日
    浏览(46)
  • IDEA项目代码飘红,但可以正常编译运行,且清除缓存也不管用,解决办法!!

    idea代码可以正常编译,运行,但是import语句飘红,鼠标放上去提示Cannot resolve symbol,找不到Java类 https://zhuanlan.zhihu.com/p/347318105 正常来说,按照上面的链接操作就可以解决了,而且大部分清除缓存就可以解决,清除缓存步骤为 : File - Invalidate Caches / Restart… - Invalidate and Res

    2024年04月17日
    浏览(36)
  • 小插曲 -- 使用Linux编写 判断程序是否在运行的小程序

    首先,在执行“ps -elf |grep xxx”时,如果xxx存在,通常会有两条结果,一个是xxx对应的PID,一个则是grep对应的PID,但是如果 我希望执行命令后,xxx存在就只有xxx对应的PID,不存在就什么都不显示的话 ,可以将指令修改成:“ ps -elf |grep XXX|grep -v grep ”,这样就可以屏蔽显示

    2024年02月07日
    浏览(47)
  • IDEA中运行JSP文件,启动Tomcat运行网页显示404,对编译代码右键运行又可以显示正常页面的解决办法

    这个问题困扰了我许久,一直以为是自己的Tomcat配置出现了问题,但实际上我Tomcat并没有出现问题,在网上看了很多大佬关于这方面问题的博客问题也没有得到解决,直到我对比了两次运行的网页url才发现了自己的问题所在。 这里选择Edit Configurations...进去 切换到Deployment,并

    2024年02月08日
    浏览(47)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包