FDEInterviews logo
MLOps & ML Engineering / 37
mediumNVIDIAAWSDatabricks

Containerize a GPU ML pipeline with Docker. How do you match CUDA to drivers and keep it reproducible?

The 'works on my GPU box, CUDA error 803 in prod' bug, solved at the source. What the host driver actually pins, why the toolkit version is fungible, and the multi-stage build that ships a lean reproducible image.

Updated Sep 2026 · Grounded in real Forward Deployed Engineer interview loops and written to a senior-engineer editorial bar.

The 'works on my GPU box, CUDA error 803 in prod' bug, solved at the source. What the host driver actually pins, why the toolkit version is fungible, and the multi-stage build that ships a lean reproducible image.

20 answers per topic instead of 10, plus saved progress and bookmarks · no cardor unlock all 523 remaining answers · ₹2,000 / $25
UP NEXT ON YOUR JOURNEY
FEDITOR'S NOTE

The screen is whether you know the container ships the CUDA toolkit but NOT the driver: the host's NVIDIA driver is exposed through the container runtime, and the only hard constraint is driver >= the toolkit's minimum. Candidates who think you 'install the driver in the Dockerfile' or who can't explain why the same image runs on two driver versions reveal they've never actually shipped GPU containers. The strong answer pins toolkit and Python deps for reproducibility while leaving the driver as the host's job, and uses multi-stage builds to keep the runtime image lean.

DISCUSSION · 0

No comments yet — be the first to share your approach.