指定cv::cuda::GpuMat创建所在的GPU卡

1. 背景

在多卡服务器运行多应用时，要将应用跑在不同的GPU上，就需要通过代码指定所运行的GPU，cv::cuda::GpuMat创建时，同样需要指定创建所对应的GPU。

2. 函数说明

cv::cuda::setDevice 是 OpenCV 中用于设置当前使用的 GPU 设备的函数。它的语法如下：

void cv::cuda::setDevice(int device)

参数 device 是一个整数，表示要选择的 GPU 设备的索引。索引从 0 开始，表示第一个 GPU 设备。

使用 cv::cuda::setDevice 函数，您可以在执行 CUDA 相关操作之前，明确指定要使用的 GPU 设备。这对于系统中存在多个 GPU 设备的情况非常有用。通过选择特定的设备，您可以确保 CUDA 操作在指定的 GPU 上执行。

3. 代码

#include <opencv2/opencv.hpp>int main() {// 设置要使用的 GPU 设备索引int deviceIndex = 1; // 选择第二个 GPU 设备// 设置要使用的 GPU 设备cv::cuda::setDevice(deviceIndex);// 创建一个 GpuMat 对象cv::cuda::GpuMat gpuMat;// 在指定的 GPU 设备上执行其他操作// ...return 0;
}