首页 Paddle框架 帖子详情
use gpu相关问题
收藏
快速回复
Paddle框架 问答深度学习 3092 1
use gpu相关问题
收藏
快速回复
Paddle框架 问答深度学习 3092 1

跑DuReader的demo,到 sudo bash run.sh bidaf_test bidaf train的步骤时报错,

I0725 02:01:09.034232 2072 Util.cpp:166] commandline: --use_gpu=True --trainer_count=1
F0725 02:01:09.417412 2072 Allocator.h:89] Check failed: ptr Fail to allocate GPU memory 38190000 bytes

可是python文件中--use_gpu default=false,也没有传入参数,不知道为啥变成True了,使用的PaddlePaddle也是CPU版本的,直接PIp install 的,不知道问题出在哪里了。

0
收藏
回复
全部评论(1)
时间顺序
挪威的仙人掌
#3 回复于2018-08

此前有小伙伴也遇到了这个问题,您可以查看如下链接中,有对应的回复

http://ai.baidu.com/forum/topic/show/870085

 

【parser.add_argument('--use_gpu', action='store_true', default=False)
这句话的意思是: 如果检测到参数中带有 --use_gpu则设置 --use_gpu的值为True, 不带这个参数则设置为False。

run.sh的train()函数在运行 run.py的时候 指定了 --use_gpu, 所以会显示为 --use_gpu=True。

如果想用cpu运行, 可以把 run.sh中train()函数的 --use_gpu删除。】

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户