Commit 3cdabdf
authored
Remove command to set cuda path and install uv (Kaggle#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 version1 parent f48df1d commit 3cdabdf
1 file changed
Lines changed: 2 additions & 12 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -19,10 +19,8 @@ RUN cat /kaggle_requirements.txt >> /requirements.txt | |||
| 19 | 19 | # TODO: GPU requirements.txt | |
| 20 | 20 | # TODO: merge them better (override matching ones). | |
| 21 | 21 | ||
| 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 | ||
| 26 | 24 | ||
| 27 | 25 | # Install manual packages: | |
| 28 | 26 | # 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 | |||
| 71 | 69 | ||
| 72 | 70 | # Install GPU-specific non-pip packages. | |
| 73 | 71 | {{ 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 | - | ||
| 82 | 72 | RUN uv pip install --system "pycuda" | |
| 83 | 73 | ||
| 84 | 74 | # b/381256047 Remove once installed in Colabs base image. | |
0 commit comments