use gpu相关问题
收藏
跑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
收藏
请登录后评论
此前有小伙伴也遇到了这个问题，您可以查看如下链接中，有对应的回复
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删除。】