paddlepaddle 查看GPU设备号,并启动单或者多GPU训练

paddle深度学习框架

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

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

 输出:

gpu:0

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

以手写字识别为例:设置0.

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

 如果你有多张GPU ,设置 0号、 1号、 2号 、3号、 4号、 5号、 6号 ....

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

你可能感兴趣的:(paddlepaddle,paddle,python,开发语言)