get_device_name

paddle.device.cuda. get_device_name ( device=None ) [源代码]

返回从 CUDA 函数 cudaDeviceProp 获取到的设备名称。

参数

  • device (paddle.CUDAPlace|int,可选) - 希望获取名称的设备或者设备 ID。如果 device 为 None(默认),则为当前的设备。

返回

str:设备的名称。

代码示例

>>> import paddle
>>> paddle.device.set_device('gpu')

>>> paddle.device.cuda.get_device_name()

>>> paddle.device.cuda.get_device_name(0)

>>> paddle.device.cuda.get_device_name(paddle.CUDAPlace(0))