首页 > 解决方案 > 如何在 Tensorflow 中获取 CUDA/GPU/设备指针?

问题描述

image使用 OpenCV(cv2)从视频文件中读取和解码。我也有一张我只用 OpenCV 读过一次的图片(mask)。我的模型读取对mask- image。如何确保掩码仅在 GPU 上移动/创建一次?如何在TensorFlow 使用的 OpenCV 解码图像上说类似.to_device()的东西?.cuda()

标签: pythontensorflowopencvgpu

解决方案


推荐阅读