查看GPU

最新推荐文章于 2025-02-18 15:32:33 发布

今天也是要加油的DD

最新推荐文章于 2025-02-18 15:32:33 发布

阅读量930

点赞数

本文链接：https://blog.youkuaiyun.com/weixin_48354103/article/details/106723215

版权

本文介绍如何查看并了解当前系统中可用的GPU资源，包括型号、内存等关键信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

查看当前可使用的GPU：

# import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
'''
Device mapping:
/job:localhost/replica:0/task:0/device:XLA_CPU:0 -> device: XLA_CPU device
/job:localhost/replica:0/task:0/device:XLA_GPU:0 -> device: XLA_GPU device'''

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: 16943220086675384530
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 1694216732785436673
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 1874818815327916504
physical_device_desc: "device: XLA_GPU device"
]
'''