get_device_properties¶

paddle.device.cuda. get_device_properties ( device ) [源代码] ¶

返回给定的设备属性。

参数¶

device (paddle.CUDAPlace or int or str) - 设备、设备 ID 和类似于 gpu:x 的设备名称。如果 device 为空，则 device 为当前的设备。默认值为 None。

返回¶

_gpuDeviceProperties：设备属性，包括标识设备的 ASCII 字符串、设备计算能力的主版本号以及次版本号、全局显存总量、设备上多处理器的数量。

代码示例¶

          >>> import paddle
>>> paddle.device.set_device('gpu')
>>> paddle.device.cuda.get_device_properties()
>>> # _gpuDeviceProperties(name='A100-SXM4-40GB', major=8, minor=0, total_memory=40536MB, multi_processor_count=108)

>>> paddle.device.cuda.get_device_properties(0)
>>> # _gpuDeviceProperties(name='A100-SXM4-40GB', major=8, minor=0, total_memory=40536MB, multi_processor_count=108)

>>> paddle.device.cuda.get_device_properties('gpu:0')
>>> # _gpuDeviceProperties(name='A100-SXM4-40GB', major=8, minor=0, total_memory=40536MB, multi_processor_count=108)

>>> paddle.device.cuda.get_device_properties(paddle.CUDAPlace(0))
>>> # _gpuDeviceProperties(name='A100-SXM4-40GB', major=8, minor=0, total_memory=40536MB, multi_processor_count=108)

         