如何从CUDA映射的OpenGL纹理中获取原始device_ptr数据？

一、可行性确认 #

完全可以在GPU内部实现OpenGL纹理（颜色、深度）到CUDA连续device_ptr的映射，替代glReadPixels完成GPU端数据读取，无需将数据回传CPU。

二、cudaArray转连续device_ptr的实现 #

cudaGraphicsSubResourceGetMappedArray返回的cudaArray是纹理优化布局（非连续内存），要得到连续device_ptr，可通过CUDA内存复制实现：

步骤1：注册并映射OpenGL纹理 #

注册时需指定cudaGraphicsRegisterFlagsSurfaceLoadStore标志，允许CUDA对资源进行读写：

cudaGraphicsResource* cuda_gl_res;
// 注册颜色纹理（示例）
cudaGraphicsGLRegisterImage(&cuda_gl_res, color_tex_id, GL_TEXTURE_2D, cudaGraphicsRegisterFlagsSurfaceLoadStore);

// 映射资源到CUDA
cudaGraphicsMapResources(1, &cuda_gl_res, nullptr);
cudaArray* cuda_arr;
cudaGraphicsSubResourceGetMappedArray(&cuda_arr, cuda_gl_res, 0, 0);

步骤2：分配连续CUDA设备内存并复制数据 #

可以用cudaMemcpy2DFromArray直接复制，无需核函数：

// 颜色纹理：GL_UNSIGNED_BYTE对应uchar4（RGBA）
uchar4* color_dev_ptr;
size_t color_pitch = width * sizeof(uchar4);
cudaMalloc(&color_dev_ptr, width * height * sizeof(uchar4));
cudaMemcpy2DFromArray(color_dev_ptr, color_pitch, cuda_arr, 0, 0, color_pitch, height, cudaMemcpyDeviceToDevice);

// 深度纹理：GL_UNSIGNED_INT对应uint
uint* depth_dev_ptr;
size_t depth_pitch = width * sizeof(uint);
cudaMalloc(&depth_dev_ptr, width * height * sizeof(uint));
cudaMemcpy2DFromArray(depth_dev_ptr, depth_pitch, cuda_arr, 0, 0, depth_pitch, height, cudaMemcpyDeviceToDevice);

此时color_dev_ptr和depth_dev_ptr就是连续的CUDA设备指针，可直接用于后续CUDA操作。

三、深度纹理注册崩溃的解决办法 #

崩溃通常源于格式不兼容，按以下步骤排查修复：

• 确保深度纹理的内部格式与数据类型匹配：使用GL_DEPTH_COMPONENT32UI作为内部格式（对应GL_UNSIGNED_INT数据类型），创建纹理代码示例：

  GLuint depth_tex;
  glGenTextures(1, &depth_tex);
  glBindTexture(GL_TEXTURE_2D, depth_tex);
  glTexImage2D(GL_TEXTURE_2D, 0, GL_DEPTH_COMPONENT32UI, width, height, 0, GL_DEPTH_COMPONENT, GL_UNSIGNED_INT, nullptr);
  // 设置过滤模式为NEAREST，避免纹理采样插值
  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
  

• 注册深度纹理时，同样使用cudaGraphicsRegisterFlagsSurfaceLoadStore标志，确保CUDA版本（8.0+）支持该格式的资源映射。
• 若仍崩溃，可通过FBO将深度纹理转换为颜色纹理（如GL_R32UI）再映射，作为兼容备选方案。

四、关键注意事项 #

• 资源映射/解映射必须成对调用：完成CUDA操作后，调用cudaGraphicsUnmapResources(1, &cuda_gl_res, nullptr)释放资源，避免OpenGL与CUDA冲突。
• 重复操作时，保留cudaGraphicsResource对象，无需每次重新注册，提升性能。
• 数据类型严格匹配：颜色纹理GL_UNSIGNED_BYTE对应uchar4，深度纹理GL_UNSIGNED_INT对应uint，避免类型错误导致数据异常。

内容的提问来源于stack exchange，提问作者qcl