Skip to content

Commit 3cdabdf

Browse files
Remove command to set cuda path and install uv (#1485)
Remove command to install uv as it is installed in the Colab base image. Remove updating cuda path as Colab base image sets the CUDA version
1 parent f48df1d commit 3cdabdf

File tree

1 file changed

+2
-12
lines changed

1 file changed

+2
-12
lines changed

Dockerfile.tmpl

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,8 @@ RUN cat /kaggle_requirements.txt >> /requirements.txt
1919
# TODO: GPU requirements.txt
2020
# TODO: merge them better (override matching ones).
2121

22-
# Install uv & Kaggle packages
23-
RUN curl -LsSf https://astral.sh/uv/install.sh | sh
24-
RUN export PATH="${HOME}/.local/bin:${PATH}" && uv pip install --system -r /requirements.txt
25-
ENV PATH="~/.local/bin:${PATH}"
22+
# Install Kaggle packages
23+
RUN uv pip install --system -r /requirements.txt
2624

2725
# Install manual packages:
2826
# b/183041606#comment5: the Kaggle data proxy doesn't support these APIs. If the library is missing, it falls back to using a regular BigQuery query to fetch data.
@@ -71,14 +69,6 @@ ARG PACKAGE_PATH=/usr/local/lib/python3.11/dist-packages
7169

7270
# Install GPU-specific non-pip packages.
7371
{{ if eq .Accelerator "gpu" }}
74-
ARG CUDA_MAJOR_VERSION \
75-
CUDA_MINOR_VERSION
76-
ENV CUDA_MAJOR_VERSION=${CUDA_MAJOR_VERSION} \
77-
CUDA_MINOR_VERSION=${CUDA_MINOR_VERSION}
78-
79-
# Make sure we are on the right version of CUDA
80-
RUN update-alternatives --set cuda /usr/local/cuda-$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION
81-
8272
RUN uv pip install --system "pycuda"
8373

8474
# b/381256047 Remove once installed in Colabs base image.

0 commit comments

Comments
 (0)