Keras multi-gpu: указание явных идентификаторов графического процессора

Question

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

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

Answer 1

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

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

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

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