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

Posted ZSYL

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了教你如何一招查看 Tensorflow 是GPU版本 or CPU版本相关的知识,希望对你有一定的参考价值。

查看 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"

即可查看!

参考link

以上是关于教你如何一招查看 Tensorflow 是GPU版本 or CPU版本的主要内容,如果未能解决你的问题,请参考以下文章

教你一招,利用 PyTorch on DC/OS 实现深度学习

没GPU也能玩梵高作画:Ubuntu tensorflow CPU版

查看TensorFlow的版本以及安装路径

windows 下 TensorFlow(GPU 版)的安装

[转]教你一招 - 如何给nopcommerce增加一个类似admin的area

在Docker中使用TensorFlow(GPU版+Ubuntu 20.04)