File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -47,8 +47,9 @@ RUN uv pip install --no-build-isolation --system "git+https://github.com/Kaggle/
4747# b/408281617: Torch is adamant that it can not install cudnn 9.3.x , only 9.1.x , but Tensorflow can only support 9.3.x.
4848# This conflict causes a number of package downgrades, which are handled in this command
4949# b/302136621: Fix eli5 import for learntools
50+ # b/416137032: cuda 12.9.0 breaks datashader 1.18.0
5051RUN uv pip install --system --force-reinstall --extra-index -url https://pypi.nvidia.com " cuml-cu12==25.2.1" \
51- " nvidia-cudnn-cu12==9.3.0.75" cuda-bindings== 12.8.0 cuda-python == 12.8.0 \
52+ " nvidia-cudnn-cu12==9.3.0.75" cuda-bindings== 12.8.0 2 == 12.8.0 \
5253 scipy tsfresh scikit-learn== 1.2.2 category-encoders eli5
5354
5455RUN uv pip install --system --force-reinstall " pynvjitlink-cu12==0.5.2"
You can’t perform that action at this time.
0 commit comments