在PaddlePaddle中指定使用GPU进行运行非常简单。首先,确保你的机器上已经安装了CUDA和cuDNN,并且正确配置了GPU环境。然后,按照以下步骤进行操作:
导入PaddlePaddle库:
import paddle
设置使用的设备为GPU:
paddle.set_device('gpu')
如果你有多块GPU,你可以选择指定使用的GPU编号:
paddle.set_device('gpu:0') # 使用第一块GPU
在你的代码中,使用PaddlePaddle提供的API进行模型训练或推理等操作。
这样,你的代码就会在GPU上运行了。