Airflow Docker build installs heavy NVIDIA CUDA dependencies via docling
Author: kshitijkumrawat20Created Jul 2, 2026Updated Jul 3, 2026
Bug
When running the Docker build for this repository, the image build process pulls several gigabytes of NVIDIA CUDA dependencies.
Root Cause
The project includes docling, which relies on torch and torchvision. When uv builds inside a Linux Docker container, it pulls the standard PyPI Linux wheels, which default to bundling full CUDA binaries. Running uv add docling locally on a Mac or Windows machine bypasses this because PyPI delivers native CPU/MPS wheels for those operating systems.
Suggested Fix
We should force the build to look at the PyTorch CPU index.
Source: jamwithai/production-agentic-rag-course