Skip to content

Commit a5de821

Browse files
committed
Upgrade Colab base image
From `20251024-060052_RC00` to `20260126-060048_RC00`. Added justification to the diff about other changes included. http://b/331651883
1 parent 7b88626 commit a5de821

3 files changed

Lines changed: 4 additions & 11 deletions

File tree

Dockerfile.tmpl

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -23,15 +23,9 @@ RUN uv pip uninstall --system --no-cache google-cloud-bigquery-storage
2323
# to avoid affecting the larger build, we'll post-install it.
2424
RUN uv pip install --no-build-isolation --no-cache --system "git+https://github.com/Kaggle/learntools"
2525

26-
# newer daal4py requires tbb>=2022, but libpysal is downgrading it for some reason
27-
RUN uv pip install --system --no-cache "tbb>=2022" "libpysal==4.9.2"
28-
29-
# b/404590350: Ray and torchtune have conflicting tune cli, we will prioritize torchtune.
30-
# b/315753846: Unpin translate package, currently conflicts with adk 1.17.0
31-
# b/468379293: Unpin Pandas once cuml/cudf are compatible, version 3.0 causes issues
32-
# b/468383498: numpy will auto-upgrade to 2.4.x, which causes issues with numerous packages
26+
# b/404590350: Ray and torchtune have conflicting cli named `tune`. `ray` is not part of Colab's base image. Re-install `tune` to ensure the torchtune CLI is available by default.
3327
# b/468367647: Unpin protobuf, version greater than v5.29.5 causes issues with numerous packages
34-
RUN uv pip install --system --force-reinstall --no-cache --no-deps torchtune "google-cloud-translate==3.12.1" "numpy==2.0.2" "pandas==2.2.2"
28+
RUN uv pip install --system --force-reinstall --no-cache --no-deps torchtune
3529
RUN uv pip install --system --force-reinstall --no-cache "protobuf==5.29.5"
3630

3731
# Adding non-package dependencies:

config.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
BASE_IMAGE=us-docker.pkg.dev/colab-images/public/runtime
2-
BASE_IMAGE_TAG=release-colab-external_20251024-060052_RC00
2+
BASE_IMAGE_TAG=release-colab-external_20260126-060048_RC00
33
CUDA_MAJOR_VERSION=12
44
CUDA_MINOR_VERSION=5

kaggle_requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,7 @@ keras-nlp
6565
keras-tuner
6666
kornia
6767
langid
68-
# b/328788268: libpysal 4.10 seems to fail with "module 'shapely' has no attribute 'Geometry'. Did you mean: 'geometry'"
69-
libpysal<=4.9.2
68+
libpysal
7069
lime
7170
line_profiler
7271
mamba

0 commit comments

Comments
 (0)