Pin nvidia-cublas-cu12 to version 12.9.1.4 in Dockerfile.ci to ensure consistent dependency management in the CI environment.

Signed-off-by: Taichang Zhou <tzhouam@connect.ust.hk>
This commit is contained in:
Taichang Zhou committed 2026-02-11 20:31:57 +08:00
1 parent 05784d3dd0
commit eb05967451
1 file changed
+3
+3
View File
@@ -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.