记录一下
使用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)
记录一下
使用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)
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
