教你如何一招查看 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())
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())
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"
[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"
[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"
即可查看!
版权声明:
作者:admin
链接:http://blog.mryxh.cn/396.html
文章版权归作者所有,未经允许请勿转载。
THE END

0

打赏

分享

二维码

海报