paddle paddle 中使用gpu训练

记录一下

使用paddle的device的get和set来获取和设置

import paddle
print(paddle.device.get_device())

如果有gpu的话运行结果如下：

之后在代码开始位置按照设备名通过set_device()方法设置设备即可：

paddle.device.set_device('gpu:0') # 把get—device的结果直接复制进去
# 之后就可以正常运行其他内容，如
model = MNIST()
train(model)