在AI studi上使用最基础的paddle-API来训练模型时怎么使用GPU

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

双双子zh00 发布于2021-08

怎么使用GPU

0

收藏

回复

全部评论(1)

三岁

#2 回复于2021-08

AI Studio可以直接运行高级版本和至尊版本。默认GPU,也可以使用代码定义。

# cpu/gpu环境选择，在 paddle.set_device() 输入对应运行设备。
# device = paddle.set_device('gpu')

0

回复

提issue

需求/bug反馈？一键提issue告诉我们

提pr

发现bug？如果您知道修复办法，欢迎提pr直接参与建设飞桨~