Commit aecfe94
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -35,6 +35,13 @@ ADD patches/kaggle_datasets.py /root/.local/lib/python3.8/site-packages/kaggle_d | |||
| 35 | 35 | ENV PIP_LIBTPU=/usr/local/lib/python3.8/site-packages/libtpu/libtpu.so | |
| 36 | 36 | ENV PYTORCH_LIBTPU=/lib/torch-libtpu.so | |
| 37 | 37 | ||
| 38 | + # Tensorflow wheel: | ||
| 39 | + RUN pip install tensorflow_hub https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/tensorflow/tf-${TENSORFLOW_VERSION}/tensorflow-${TENSORFLOW_VERSION}-cp38-cp38-linux_x86_64.whl | ||
| 40 | + RUN curl --output /lib/libtpu.so https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/libtpu/${TF_LIBTPU_VERSION}/libtpu.so | ||
| 41 | + ADD patches/kaggle_module_resolver.py /usr/local/lib/python3.8/site-packages/tensorflow_hub/kaggle_module_resolver.py | ||
| 42 | + RUN sed -i '/from tensorflow_hub import uncompressed_module_resolver/a from tensorflow_hub import kaggle_module_resolver' /usr/local/lib/python3.8/site-packages/tensorflow_hub/config.py | ||
| 43 | + RUN sed -i '/_install_default_resolvers()/a \ \ registry.resolver.add_implementation(kaggle_module_resolver.KaggleFileResolver())' /usr/local/lib/python3.8/site-packages/tensorflow_hub/config.py | ||
| 44 | + | ||
| 38 | 45 | # Install Pytorch & related packages | |
| 39 | 46 | # https://cloud.google.com/tpu/docs/pytorch-xla-ug-tpu-vm#changing_pytorch_version | |
| 40 | 47 | # The URL doesn't include patch version. i.e. must use 1.11 instead of 1.11.0 | |
@@ -55,13 +62,6 @@ RUN pip install jax[tpu]==${JAX_VERSION} -f https://storage.googleapis.com/jax-r | |||
| 55 | 62 | https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/tensorflow/tf-${TENSORFLOW_VERSION}/tensorflow-${TENSORFLOW_VERSION}-cp38-cp38-linux_x86_64.whl \ | |
| 56 | 63 | numpy==1.23.5 | |
| 57 | 64 | ||
| 58 | - # Tensorflow wheel: | ||
| 59 | - RUN pip install --force-reinstall tensorflow_hub https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/tensorflow/tf-${TENSORFLOW_VERSION}/tensorflow-${TENSORFLOW_VERSION}-cp38-cp38-linux_x86_64.whl | ||
| 60 | - RUN curl --output /lib/libtpu.so https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/libtpu/${TF_LIBTPU_VERSION}/libtpu.so | ||
| 61 | - ADD patches/kaggle_module_resolver.py /usr/local/lib/python3.8/site-packages/tensorflow_hub/kaggle_module_resolver.py | ||
| 62 | - RUN sed -i '/from tensorflow_hub import uncompressed_module_resolver/a from tensorflow_hub import kaggle_module_resolver' /usr/local/lib/python3.8/site-packages/tensorflow_hub/config.py | ||
| 63 | - RUN sed -i '/_install_default_resolvers()/a \ \ registry.resolver.add_implementation(kaggle_module_resolver.KaggleFileResolver())' /usr/local/lib/python3.8/site-packages/tensorflow_hub/config.py | ||
| 64 | - | ||
| 65 | 65 | # Packages needed by the Notebook editor: | |
| 66 | 66 | RUN pip install papermill jupyterlab python-lsp-server[all] jupyterlab-lsp | |
| 67 | 67 | ||
0 commit comments