Skip to content

Specifying cuda device through SupervisedRunner doesn't get picked up #478

@the-lay

Description

@the-lay

When you specify cuda device in SupervisedRunner (and SupervisedWandbRunner), it doesn't influence what GPUs will be used. Manually setting CUDA_VISIBLE_DEVICES works though.

Example

import torch
from catalyst.dl.runner import SupervisedRunner, SupervisedWandbRunner
...
runner = SupervisedRunner(device=torch.device('cuda:1'))
runner.train(...)

Expected behavior: Only GPU with ID 1 should be used.
Actual behavior: Catalyst uses all available devices.

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions