Выбор 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