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

1. 背景

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

2. 函数说明

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

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

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

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

3. 代码

#include 

int main() {
    // 设置要使用的 GPU 设备索引
    int deviceIndex = 1; // 选择第二个 GPU 设备

    // 设置要使用的 GPU 设备
    cv::cuda::setDevice(deviceIndex);

    // 创建一个 GpuMat 对象
    cv::cuda::GpuMat gpuMat;

    // 在指定的 GPU 设备上执行其他操作
    // ...

    return 0;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17