Commit 6d08572
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -13,8 +13,11 @@ ADD patches/template_conf.json /opt/kaggle/conf.json | |||
| 13 | 13 | # Tensorflow wheel: | |
| 14 | 14 | # When tensorflow is compatible with being installed alongside JAX/Pytorch then we no longer need to include the wheel and can install it directly. | |
| 15 | 15 | # RUN pip install https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/tensorflow/tf-${TENSORFLOW_VERSION}/tensorflow-${TENSORFLOW_VERSION}-cp38-cp38-linux_x86_64.whl | |
| 16 | + # To install offline: | ||
| 17 | + # pip install --no-index --find-links /lib/wheels/sources /lib/wheels/tensorflow-2.9.1-cp38-cp38-linux_x86_64.whl | ||
| 16 | 18 | RUN mkdir -p /lib/wheels && curl --output /lib/wheels/tensorflow-${TENSORFLOW_VERSION}-cp38-cp38-linux_x86_64.whl https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/tensorflow/tf-${TENSORFLOW_VERSION}/tensorflow-${TENSORFLOW_VERSION}-cp38-cp38-linux_x86_64.whl | |
| 17 | 19 | RUN curl --output /lib/libtpu.so https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/libtpu/1.3.0/libtpu.so | |
| 20 | + RUN pip download -d /lib/wheels/sources /lib/wheels/tensorflow-${TENSORFLOW_VERSION}-cp38-cp38-linux_x86_64.whl | ||
| 18 | 21 | ||
| 19 | 22 | # LIBTPU installed here: | |
| 20 | 23 | ENV PIP_LIBTPU=/usr/local/lib/python3.8/site-packages/libtpu/libtpu.so | |
0 commit comments