Select which gpu to train on about super-gradients HOT 8 CLOSED

os.environ['CUDA_VISIBLE_DEVICES'] = "1"

This solved it, thank you!

from super-gradients.

Already tried, but it's giving me a KeyError: "Trainer does not accept anymore 'device' and 'multi_gpu' as argument. Both should instead be passed to super_gradients.setup_device(device=..., multi_gpu=..., num_gpus=...)"

Trainer(device=...) is legacy and will soon be removed from SG.

The same can be done with

from super_gradients.training.utils.distributed_training_utils import setup_device

setup_device(device=...)

But this is only useful if you want to train on cpu, or with multiple gpus (DP/DDP). The default behavior is single GPU so you don't need to call it in your case

from super-gradients.

Join the discussion on DagsHub!

from super-gradients.

export CUDA_VISIBLE_DEVICES=0 python -m super_gradients.train_from_recipe ...

from super-gradients.

Hi. Is there a way to do it without using a recipe or through trainer.train?

from super-gradients.

You can always pass target device to instance of Trainer manually if you want:
trainer = Trainer(..., device = "cuda:1")

from super-gradients.

Already tried, but it's giving me a KeyError: "Trainer does not accept anymore 'device' and 'multi_gpu' as argument. Both should instead be passed to super_gradients.setup_device(device=..., multi_gpu=..., num_gpus=...)"

from super-gradients.

You can set it with env variables at the beginning of your script

import os
from super_gradients import Trainer

os.environ['CUDA_VISIBLE_DEVICES'] = "1"


# Unchanged
trainer = Trainer(...)
trainer.train(...)

from super-gradients.