From 79b7bd40c2899bad80bff13c23508220ad6ef2e9 Mon Sep 17 00:00:00 2001 From: Dustin H Date: Mon, 14 Aug 2023 14:24:41 -0400 Subject: [PATCH] Add TPU deps from PyTorch example https://www.kaggle.com/code/wcromar/pytorch-xla-2-0-on-kaggle/notebook?scriptVersionId=124841632 --- tpu/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tpu/Dockerfile b/tpu/Dockerfile index 1e56da16..bf3193d6 100644 --- a/tpu/Dockerfile +++ b/tpu/Dockerfile @@ -52,7 +52,7 @@ RUN pip install tensorflow_hub https://storage.googleapis.com/cloud-tpu-tpuvm-ar jax[tpu]==${JAX_VERSION} -f https://storage.googleapis.com/jax-releases/libtpu_releases.html trax flax optax elegy git+https://github.com/deepmind/dm-haiku jraph distrax \ numpy==1.23.5 \ papermill jupyterlab python-lsp-server[all] "jupyter-lsp==1.5.1" \ - pandas matplotlib opencv-python-headless librosa scikit-learn + pandas matplotlib opencv-python-headless librosa scikit-learn accelerate diffusers transformers # Tensorflow libtpu: RUN curl --output /lib/libtpu.so https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/libtpu/${TF_LIBTPU_VERSION}/libtpu.so