CUDA流捕获结束失败与同步报错问题求助

错误原因解析 #

1. CUDA_ERROR_STREAM_CAPTURE_UNJOINED #

该错误说明捕获流存在未合并的依赖分支——比如通过cuStreamWaitEvent或事件关联创建的辅助流，在结束捕获前没有将依赖合并回主捕获流。CUDA 12.x对分支合并的校验比旧版本更严格，再加上你当前CUDA toolkit（12.6.85）与驱动（535.54.03，对应CUDA 12.2）版本不匹配，进一步放大了兼容性问题。

2. CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED #

流捕获期间绝对禁止调用同步操作，cuStreamSynchronize()属于明确不支持的操作。旧版本CUDA可能未严格校验这类违规操作，但12.x版本强化了规则，直接抛出错误。

修复方案 #

解决未合并分支问题 #

• 梳理捕获流的依赖链：所有通过事件关联的辅助流，必须在cuStreamEndCapture()前，通过cuStreamWaitEvent将主捕获流与辅助流的完成事件绑定，完成分支合并。
• 核心修正代码示例（基于自定义API封装逻辑）：

  // 假设存在辅助流aux_stream和同步事件event
  cuEventRecord(event, aux_stream);
  cuStreamWaitEvent(capture_stream, event, 0); // 将辅助流的执行结果同步回捕获流
  cuStreamEndCapture(capture_stream, &graph);
  

解决同步不支持问题 #

• 移除捕获过程中的cuStreamSynchronize()调用：流捕获是命令序列的记录过程，不需要实时同步执行。如果需要确保捕获前的操作完成，应在启动流捕获之前调用同步，而非捕获期间。

版本兼容性修复建议 #

CUDA 12.6要求最低驱动版本为550.39.01，你当前使用的驱动（535.54.03）仅对应CUDA 12.2，跨版本差异过大必然导致未定义行为。建议二选一：

• 升级驱动至CUDA 12.6匹配的版本；
• 降级CUDA toolkit至12.2版本，保持与驱动版本一致。

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