教你如何一招查看 Tensorflow 是GPU版本 or CPU版本
2023-03-07 09:47:29 时间
查看 Tensorflow 是GPU还是CPU版本
在Python环境中输入:
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())
[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版本
相关文章
- 如何管理高度可扩展系统中的元数据?
- 静态分析工具Clang Static Analyzer (4) Clang-Tidy
- Linux 6.1 正式发布:带来 MGLRU、初始 Rust 支持
- 不良数据会造成更严重后果的八个原因
- 在 Windows 上安装 PySpark 保姆级教程
- 如何在 Linux 中找到一个进程 ID 并杀死它
- 为什么你要在 Linux 上尝试 Nemo 文件管理器?
- 在 Linux 上获取命令行帮助的七种方法
- 面试突击:了解 HTTP 协议吗?
- 大数据如何通过销售自动化创造更多的机会?
- Windows11 学院:如何修复 Microsoft Store 的 0x80070483 错误
- React:不要动,否则你会被炒鱿鱼
- 为什么云原生数据湖值得关注?
- 微软宣布将结束对 Windows 7、8 和 8.1版 Edge 浏览器的支持
- 电脑技巧:Microsoft Edge浏览器技巧介绍,值得收藏!
- Vite 4.0 正式发布!有哪些更新你知道吗?
- 让我们一起分析 Go 语言逃逸
- 静态分析工具Clang Static Analyzer (3) Cppcheck
- “低代码”十问,你能回答吗?
- CPU100%,怎么快速定位?