Paddle：指定GPU

1：查看默认运行的device设备

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

2：指定GPU，通过set_device函数接口

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

示例：

import paddle

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

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

参考链接

1： 模型训练、评估与推理-使用文档-PaddlePaddle深度学习平台

2： 分布式训练-使用文档-PaddlePaddle深度学习平台