Выбор GPU / CPU

Чтобы запустить программу на конкретном GPU, вы можете использовать переменную окружения CUDA_VISIBLE_DEVICES. Например, для запуска программы на втором GPU используйте:

CUDA_VISIBLE_DEVICES=1 python my_program.py

Чтобы запустить программу на нескольких GPU, используйте:

CUDA_VISIBLE_DEVICES=0,1 python my_program.py

Чтобы отключить использование GPU (использовать CPU), используйте:

CUDA_VISIBLE_DEVICES="" python my_program.py