diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index bd7ccd5fd..1f5fb28f6 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -20,6 +20,9 @@ RUN apt-get update && \ # causing it to fall back to the older stable release from PyPI. RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm --extra-index-url https://wheels.vllm.ai/4a1550d22d7058e129d0e1257e726b3bf4a77025 +# Pin nvidia-cublas-cu12 to a specific version +RUN uv pip install --system --upgrade --force-reinstall "nvidia-cublas-cu12==12.9.1.4" + # # Keep flashinfer runtime packages in version sync. # # We derive the version from installed flashinfer (pulled by vllm) # # and enforce the same version for flashinfer-cubin / flashinfer-jit-cache.