[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