Python Caffe cpu и режим gpu одновременно

Question

Возможно ли запустить Caffe в режиме CPU и GPU? У меня несколько моделей Caffe, но мои ресурсы графического процессора ограничены, поэтому я не могу поместить все модели в память GPU. Я хочу использовать, например. 3 модели с режимом GPU и 2 модели с режимом CPU, но команды set_mode_cpu() и set_mode_gpu() просто переключают режим для всей библиотеки.

Answer 1

На мой взгляд, вы можете написать несколько сценариев python по одному для каждой задачи. В каждом скрипте вы можете выбрать, использовать ли CPU или GPU (и устройство GPU). Затем вы можете запускать эти несколько сценариев одновременно. Но запустите несколько задач на одной карте графического процессора, что сильно замедлит скорость в моем опыте. Удачи!