PyTorch 和 CUDA 可复现性指南更新,加入 CUBLAS_WORKSPACE_CONFIG

StasBekman · x · 2026-07-21

这条帖子更新了 **PyTorch/CUDA 程序的可复现性** 说明,并配了示例代码,核心是在最近一轮变更后,确定性训练/推理需要注意更多细节。 要点包括: - 在 CUDA 上下文创建前设置 `CUBLAS_WORKSPACE_CONFIG` - 启用 `torch.use_deterministic_algorithms(True)` - 关闭 `torch.backends.cudnn.benchmark` - 同时给 Python、NumPy、`torch.manual_seed` 和 `torch.cuda.manual_seed_all` 设种子 作者强调,PyTorch/CUDA 的 determinism 规则一直在演进,因此工程文档也需要同步更新。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →