PaddleX icon indicating copy to clipboard operation
PaddleX copied to clipboard

Predictor类初始化参数新增gpu_mem: GPU显存大小

Open angel-yi opened this issue 3 years ago • 5 comments

Predictor类初始化参数新增gpu_mem: GPU显存大小

angel-yi avatar Jul 26 '22 12:07 angel-yi

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.

CLAassistant avatar Jul 26 '22 12:07 CLAassistant

HI @angel-yi 函数参数新增时,从最后面开始追加,避免对已有调用此函数的代码带来影响

jiangjiajun avatar Jul 27 '22 05:07 jiangjiajun

好的,谢谢,是我大意了

angel-yi avatar Jul 27 '22 07:07 angel-yi

另外在默认值的选取上,为什么是设定的200呢

jiangjiajun avatar Jul 28 '22 06:07 jiangjiajun

    if use_gpu:
        # 设置GPU初始显存(单位M)和Device ID
        config.enable_use_gpu(200, gpu_id)

angel-yi avatar Jul 28 '22 08:07 angel-yi