Skip to content

How to setting to use GPU? #51

@MichaelCong

Description

@MichaelCong

I have use the "os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3", but the traing still use CPU. How to setting to use GPU?
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3"

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions