[mmaction2 вызывает указанный GPU] mmaction2 использует указанное количество GPU для обучения модели

Оригинальная ссылка:blog.CSDN.net/whiff E YF/AR…

Если на моем компьютере несколько графических процессоров, как вызвать указанный графический процессор в mmaction2?

Ниже показана строка кода для достижения

tools/train.py

Перейдите в /mmaction2/tools/train.py. Найдите следующий код:

Закомментируйте код в красном поле и добавьте следующий код ниже:

cfg.gpu_ids = range(1,2)

Эта строка означает rang (1, 2), тогда необязательным является только 1, затем вызывается первый GPU (обратите внимание, что номер GPU начинается с 0). Если вы хотите вызвать второй GPU, измените его на range(2,3)

Тогда это непосредственно обучение:

Посмотрите результат: