如何在gpu上进行深度学习模型的cv2操作？

Question

cv2在对深度学习模型执行推理时正在使用CPU我想对深度学习模型使用gpu进行推理吗？我尝试做cv2.UMat（），但是仅当我不使用深度学习时才支持它，而当我开始使用深度学习时它不起作用。请让我知道我们当时如何进行cv2操作和深度学习。

Answer 1

Tensorflow / Keras（带有tensorflow后端）保留整个gpu内存，即使它们仅需要一部分内存。如果可能只保留其中一部分，则必须检查文档。