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())

posted @ 2024-12-29 11:52 lambertlt 阅读(16) 评论(0) 编辑收藏举报

刷新页面返回顶部

登录后才能查看或发表评论，立即登录或者逛逛博客园首页

相关博文：

· Mac输入 python 打开 python3

· mac m1 安装ffmpeg，配置环境变量

· Macbook Pro M1 max Apple Silicon MacOSX 13.4.1 安装Tensorflow

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

· 事实胜于雄辩,苹果MacOs能不能玩儿机器/深度(ml/dl)学习(Python3.10/Tensorflow2)

阅读排行：
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型，支持深度思考和联网搜索！
· 基于 Docker 搭建 FRP 内网穿透开源项目（很简单哒）
· 25岁的心里话
· ollama系列01：轻松3步本地部署deepseek，普通电脑可用
· 按钮权限的设计及实现

历史上的今天：
2021-12-29 windows 电脑修改hosts
2021-12-29 Java开发公司常用总结

公告

昵称： lambertlt
园龄： 4年11个月
粉丝： 18
关注： 3

+加关注

2025年3月

日

一

二

三

四

五

六

lambert's blog

停杯投箸不能食，拔剑四顾心茫然

python 验证macbook的 mps gpu是否存在

公告

搜索

我的标签

阅读排行榜

	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())