Commit abc385c
authored
Trying Keras 3 w/ tf2.16 again (Kaggle#1390)
It looks like it was actually the JAX version bump that was breaking
things, so trying the TF & Keras bump.1 parent 61bd309 commit abc385c
2 files changed
Lines changed: 4 additions & 8 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -65,16 +65,12 @@ RUN pip install tensorflow_hub https://storage.googleapis.com/cloud-tpu-tpuvm-ar | |||
| 65 | 65 | papermill jupyterlab python-lsp-server[all] "jupyter-lsp==1.5.1" \ | |
| 66 | 66 | pandas matplotlib opencv-python-headless librosa accelerate diffusers scikit-learn transformers \ | |
| 67 | 67 | seaborn timm albumentations einops pyarrow fastparquet opencv-python \ | |
| 68 | + "keras>3" keras-cv keras-nlp \ | ||
| 68 | 69 | kagglehub && \ | |
| 69 | 70 | /tmp/clean-layer.sh | |
| 70 | 71 | ||
| 71 | - # b/318672157 Install Keras 3 for loading models. | ||
| 72 | - # We ignore dependencies for now since tf2.15 should work despite pip saying it won't. | ||
| 73 | - RUN pip install --no-dependencies "keras>3" keras-cv keras-nlp && \ | ||
| 74 | - /tmp/clean-layer.sh | ||
| 75 | - | ||
| 76 | 72 | # Tensorflow libtpu: | |
| 77 | - RUN curl --output /lib/libtpu.so https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/libtpu/${TF_LIBTPU_VERSION}/libtpu.so | ||
| 73 | + RUN curl --output /usr/local/lib/python3.10/site-packages/libtpu/libtpu.so https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/libtpu/${TF_LIBTPU_VERSION}/libtpu.so | ||
| 78 | 74 | ||
| 79 | 75 | # Kaggle Model Hub patches: | |
| 80 | 76 | ADD patches/kaggle_module_resolver.py /usr/local/lib/${PYTHON_VERSION_PATH}/site-packages/tensorflow_hub/kaggle_module_resolver.py | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -3,8 +3,8 @@ PYTHON_WHEEL_VERSION=cp310 | |||
| 3 | 3 | PYTHON_VERSION_PATH=python3.10 | |
| 4 | 4 | # gsutil ls gs://cloud-tpu-tpuvm-artifacts/tensorflow | |
| 5 | 5 | # https://cloud.google.com/tpu/docs/supported-tpu-configurations#libtpu_versions | |
| 6 | - TENSORFLOW_VERSION=2.15.0 | ||
| 7 | - TF_LIBTPU_VERSION=1.9.0 | ||
| 6 | + TENSORFLOW_VERSION=2.16.1 | ||
| 7 | + TF_LIBTPU_VERSION=1.10.1 | ||
| 8 | 8 | TF_LINUX_WHEEL_VERSION=manylinux_2_17_x86_64.manylinux2014_x86_64 | |
| 9 | 9 | JAX_VERSION=0.4.23 | |
| 10 | 10 | # gsutil ls gs://pytorch-xla-releases/wheels/tpuvm/* | grep libtpu | grep -v -E ".*rc[0-9].*" | |
0 commit comments