如何在图形处理器上使用深度学习模型进行cv2操作？

Question

cv2在深度学习模型上进行推理时使用的是图形处理器，我想使用gpu进行深度学习模型的推理，有什么方法吗？我尝试做cv2.UMat()，但它只在我不使用深度学习时才支持，当我开始使用深度学习时，它不起作用。请让我知道我们如何做cv2操作和深度学习的时候。

Answer 1

Tensorflow/Keras (使用tensorflow后端)保留整个gpu内存，即使它们只需要部分内存。如果有可能只预定一部分的话，你得检查一下文档。