Keras multi-gpu: указание явных идентификаторов GPU

Глядя на файл keras/utils/multi_gpu_utils.py в репозитории tensorflow GitHub, я увидел, что, учитывая, что вы указали, что хотите использовать x GPU, он автоматически выделит идентификаторы GPU из range(x), то есть 0, 1, 2, ..., x - 1.

Мне нужно использовать GPU 4, 5, 6 ,7, так как первые 4 GPU уже работают над другой задачей. Есть ли способ указать это?


keras tensorflow multi-gpu
person SomethingSomething    schedule 03.06.2019    source источник

Ответы (1)


arrow_upward
1
arrow_downward

В питоне вы можете использовать

import os
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"

Или установите CUDA_VISIBLE_DEVICES=0,1 в bash перед запуском скрипта python

Вы также можете обратиться к моему ответу здесь, чтобы автоматизировать этот процесс.

person y.selivonchyk    schedule 03.06.2019
comment
Спасибо. звучит просто и удивительно. Я сообщу вам, как только попробую - person SomethingSomething; 03.06.2019
comment
Это действительно сработало. Спасибо за простой и очень полезный ответ. Обратите внимание, что я использовал export CUDA_VISIBLE_DEVICES=4,5,6,7 в соответствии с доступностью моего графического процессора. - person SomethingSomething; 05.06.2019