Укажите GPU для запуска обучения (API обнаружения объектов TF, модель zoo)

Question

Я обнаруживаю объект обучения на устройстве с несколькими графическими процессорами, и хочу запустить обучение на GPU 1 (оставляя свободными 0 и 2) и не вижу возможности сделать это в начале обучения. Я просмотрел train.py и model_main.py и не могу найти строку для изменения там же. Есть предложения?

Answer 1

использовать переменную окружения CUDA_VISIBLE_DEVICES. Вы можете сделать это, добавив в вашу командную строку следующее:

CUDA_VISIBLE_DEVICES=1 python <your-python-script>

это позволит включить только графический процессор 1 для TF.