Skip to content

How to use all GPUs for training #10

@dreamstorm25

Description

@dreamstorm25

Hi, I have been trying to train on 2 gpus but I dont know how to specify both the GPUs. In the help it says if "all" then try to allocate on every gpu but using 'all' gives an error : ValueError : invalid literal for int() with base 10: all. Also specifying two numbers doesn't work. Can you please tell me how to use multiple GPUs ?

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions