Как я могу выполнять операции cv2 вдоль боковой модели глубокого обучения на GPU?

Question

cv2 использует процессор при выполнении логического вывода для модели глубокого обучения. Я хочу сделать логический вывод, используя gpu для моделей глубокого обучения. Есть ли способ? Я пытался сделать cv2.UMat (), но он поддерживал только тогда, когда я не использую глубокое обучение, когда я начал использовать глубокое обучение, это не сработало. Пожалуйста, дайте мне знать, как мы можем выполнять операции cv2 и глубокое обучение в то время.

Answer 1

Tensorflow / Keras (с бэкэндом тензорного потока) резервирует всю память графического процессора, даже если им нужна только ее часть. Вам нужно будет проверить документы, если можно зарезервировать только их часть.