Reintroduce installation of flashinfer-jit-cache in Dockerfile.ci to enhance dependency management for the CI environment, ensuring compatibility with the latest requirements.
Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
1 parent
a17078974f
commit
4d8a76e721
1 file changed
+3
@@ -25,6 +25,9 @@ RUN uv pip install --system --upgrade \
|
||||
"flashinfer-cubin==0.6.4" \
|
||||
"nvidia-cublas-cu12==12.9.1.4" \
|
||||
"numpy==2.2.6"
|
||||
RUN uv pip install --system --upgrade \
|
||||
"flashinfer-jit-cache==0.6.4" \
|
||||
--index-url https://flashinfer.ai/whl/cu129
|
||||
RUN ln -sf /usr/bin/python3 /usr/bin/python
|
||||
|
||||
ENTRYPOINT []
|
||||
Reference in new issue
Block a user