Skip to content

How to train with single GPU? #27

@yaoxinreaps

Description

@yaoxinreaps

作者你好,在accelerate_train.py中看到默认使用的是当前全部可用的GPU,怎么设置使其只用单个GPU呢?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    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