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)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值