Paddle: Specified GPU

1: Check the default running device

    import paddle
    print(paddle.device.get_device()) # 默认是GPU0

2: Specify the GPU through the set_device function interface

    import paddle
    print(paddle.device.set_device('gpu:1'))
    print(paddle.device.get_device())

Example:

import paddle

# 指定在 CPU 上训练
paddle.device.set_device('cpu')

# 指定在 GPU 第 0 号卡上训练
# paddle.device.set_device('gpu:0')

Reference link

1:  Model training, evaluation and inference-usage documentation-PaddlePaddle deep learning platform

2:  Distributed training-usage documentation-PaddlePaddle deep learning platform