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:
1 parent
05784d3dd0
commit
eb05967451
1 file changed
+3
@@ -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.
|
||||
|
||||
Reference in new issue
Block a user