Репост от Воутера Дж.
Как я могу запустить потоковое обучение на графическом процессоре (ах)?
В настоящее время я пытаюсь заставить ppo_runner тест работать с двумя моими графическими процессорами. Однако я не могу найти способ использовать эти ресурсы. Что мне нужно изменить в коде, чтобы использовать графический процессор?
Репост от команды Flow
Возможно, для начала стоит научиться использовать несколько графических процессоров через Ray:
https://ray.readthedocs.io/en/latest/using-ray-with-gpus.html
Изучение того, как использовать несколько графических процессоров с Ray, также позволит вам тренироваться с несколькими графическими процессорами, потому что мы используем Ray RLlib для обучения. Хотя справедливое предупреждение: если ваша проблема не подходит для графических процессоров (ввод больших изображений, большие нейронные сети), часто обучение на многопоточном процессоре (например, 64-поточном Xeon) намного быстрее, чем обучение на нескольких графических процессорах.
