get_available_device

paddle.device. get_available_device ( ) [源代码]

该功能获得所有可用的设备。

返回

返回列表包含所有可用的设备。

代码示例

import paddle
paddle.device.get_available_device()

# Case 1: paddlepaddle-cpu package installed, and no custom device registerd.
# Output: ['cpu']

# Case 2: paddlepaddle-gpu package installed, and no custom device registerd.
# Output: ['cpu', 'gpu:0', 'gpu:1']

# Case 3: paddlepaddle-cpu package installed, and custom deivce 'CustomCPU' is registerd.
# Output: ['cpu', 'CustomCPU']

# Case 4: paddlepaddle-gpu package installed, and custom deivce 'CustomCPU' and 'CustomGPU' is registerd.
# Output: ['cpu', 'gpu:0', 'gpu:1', 'CustomCPU', 'CustomGPU:0', 'CustomGPU:1']