Updated 2026-06 · first seen 2019-01 · recurring
PyTorch wheel ≠ installed CUDA
Container has CUDA 12.x drivers but the torch build expects another toolkit - imports work, .cuda() explodes at runtime.
- pytorch
- cuda
- What breaks
- GPU training/inference jobs die on first tensor move to device.
- How to detect
- python -c "import torch; print(torch.version.cuda, torch.cuda.is_available())" and nvidia-smi in the same image.
- Mitigation
- Pin torch index URL to the CUDA build you actually ship; bake and test the image on a GPU runner before promote.