paddle 使用GPU

Longer404

于 2021-09-29 20:47:03 发布

阅读量1.7w

点赞数 9

文章标签： paddle 深度学习

本文链接：https://blog.youkuaiyun.com/qq_31034951/article/details/120556039

版权

本文档介绍了如何在PaddlePaddle中利用GPU进行训练。首先，通过`paddle.device.get_device()`检查系统可用设备，若存在GPU，则结果将显示GPU信息。接着，在代码开头使用`paddle.device.set_device('gpu:0')`设置设备为GPU。之后，可以正常初始化模型并进行训练，例如创建MNIST模型并进行训练。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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)