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)