From 10da45636f1850dd241038ce86ac2929567a2d80 Mon Sep 17 00:00:00 2001 From: Dustin H Date: Thu, 25 Apr 2024 13:54:46 -0400 Subject: [PATCH] Fix libtpu.so location for tf2.16 --- tpu/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tpu/Dockerfile b/tpu/Dockerfile index fe79260f..ed9040a8 100644 --- a/tpu/Dockerfile +++ b/tpu/Dockerfile @@ -70,7 +70,7 @@ RUN pip install tensorflow_hub https://storage.googleapis.com/cloud-tpu-tpuvm-ar /tmp/clean-layer.sh # Tensorflow libtpu: -RUN curl --output /lib/libtpu.so https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/libtpu/${TF_LIBTPU_VERSION}/libtpu.so +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 # Kaggle Model Hub patches: ADD patches/kaggle_module_resolver.py /usr/local/lib/${PYTHON_VERSION_PATH}/site-packages/tensorflow_hub/kaggle_module_resolver.py