Updated PyTorch and CUDA reproducibility guidance adds CUBLAS_WORKSPACE_CONFIG
StasBekman · x · 2026-07-21
An ML engineering note updates the reproducibility section for PyTorch/CUDA programs.
Key points in the code example:
- set CUBLASWORKSPACECONFIG before CUDA is initialized
- call torch.usedeterministicalgorithms(True)
- disable torch.backends.cudnn.benchmark
- seed Python, NumPy, torch.manualseed, and torch.cuda.manualseedall
The post highlights that deterministic behavior in PyTorch/CUDA has continued to evolve, and the documentation now reflects those newer requirements.
More from Infra
- China’s AI arms race is increasingly defined by chips, data centers, and open models — BenBajarin · 2026-07-22
- Agent search bottlenecks are now about variance, not raw latency — rohanpaul_ai · 2026-07-22
- Gavin Baker argues Nvidia may be one of open source AI’s biggest supporters — GavinSBaker · 2026-07-22
- AI Power Demand Exposes US Energy Gap, Urging Shift from Scarcity to Abundance — bradneuberg · 2026-07-22
- Gavin Baker says Nvidia’s $630B figure would be system revenue, not all Nvidia’s — GavinSBaker · 2026-07-22
- A Firecracker-based platform says it can host 6,000 AI agents on one 256 GB server — maritime_sh · 2026-07-22