paddle怎么切换cpu和gpu版本啊

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

所所谓丶Vea 发布于2021-11

基于ResNet50_vd模型训练** - **脚本：**Python tools/train.py -c ./configs/quick_start/ResNet50_vd.yaml

电脑上有gpu和cpu版本，但是不知道怎么切换成gpu的。

全部评论(5)

所

所谓丶Vea

#2 回复于2021-11

还是说得把cpu的版本卸了？

三岁

#3 回复于2021-11

添加有关设置代码就行了

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

深渊上的坑

#4 回复于2021-11

paddledetection有切换开关的，看下输入参数里哪里要写GPU的，改一下就对了

yanshao2011

#6 回复于2022-07

搞定了么？在哪里配置

Marco809

#7 回复于2022-07

Set your default device as gpu, plz read the relevant official docs

提issue

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

提pr

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