#2382·cocoindex

[BUG] 子进程 GPU 模式丢失池的 GPU 分配

作者: NithinGoud2605创建于 2026年9月7日更新于 2026年9月9日

def probe(tag: str) -> dict: """Runs under the GPU runner; reports the GPU it can actually see.""" info = { "tag": tag, "pid": os.getpid(), "gpu": coco.current_gpu(), "cvd": os.environ.get("CUDA_VISIBLE_DEVICES"), } time.sleep(HOLD) # hold the GPU so all N calls overlap return info

内容来源: cocoindex-io/cocoindex