Skip to content
This repository was archived by the owner on Jan 24, 2024. It is now read-only.
This repository was archived by the owner on Jan 24, 2024. It is now read-only.

cpu和gpu同时使用问题 #88

Open
Open
cpu和gpu同时使用问题#88
@Echo0117

Description

@Echo0117
Echo0117
opened on Sep 24, 2020

我已经在gpu上训练完了基于palm的模型，预测的话可以实现请问可以实现在cpu和gpu上都能运行吗？
我把palm依赖于paddle关于cuda的代码注释掉是可以在cpu上跑的。但是要实现cpu和gpu同时运行（留一个use_gpu=True/False)的属性，是需要对paddle.fluid.core_avx'的 'get_cuda_device_count进行修改吗

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions