目录
- 1. 背景
- 2. 函数说明
- 3. 代码
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;
}