前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >教你如何一招查看 Tensorflow 是GPU版本 or CPU版本

教你如何一招查看 Tensorflow 是GPU版本 or CPU版本

作者头像
超级小可爱
发布2023-02-20 11:46:41
1.9K0
发布2023-02-20 11:46:41
举报
文章被收录于专栏:小孟开发笔记

查看 Tensorflow 是GPU还是CPU版本

Python环境中输入:

代码语言:javascript
复制
import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"

if __name__ == "__main__":
    print(device_lib.list_local_devices())
代码语言:javascript
复制
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 1200794696603410792
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 2454712320
locality {
  bus_id: 1
  links {
  }
}
incarnation: 15293349108358879578
physical_device_desc: "device: 0, name: NVIDIA Tegra X2, pci bus id: 0000:00:00.0, compute capability: 6.2"

即可查看!

未经允许不得转载:肥猫博客 » 教你如何一招查看 Tensorflow 是GPU版本 or CPU版本

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 查看 Tensorflow 是GPU还是CPU版本
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档