Skip to content

Commit 615feae

Browse files
committed
cuda fix
1 parent 131b5a1 commit 615feae

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

Dockerfile.tmpl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,8 @@ RUN uv pip install --no-build-isolation --system "git+https://github.com/Kaggle/
4848
# This conflict causes a number of package downgrades, which are handled in this command
4949
# b/302136621: Fix eli5 import for learntools
5050
RUN 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" scipy tsfresh scikit-learn==1.2.2 category-encoders eli5
51+
"nvidia-cudnn-cu12==9.3.0.75" cuda-bindings==12.8.0 cuda-python==12.8.0 \
52+
scipy tsfresh scikit-learn==1.2.2 category-encoders eli5
5253

5354
RUN uv pip install --system --force-reinstall "pynvjitlink-cu12==0.5.2"
5455

0 commit comments

Comments
 (0)