Fix libtpu.so location for tf2.16

Kaggle · Apr 25, 2024 · 10da456 · 10da456
1 parent 217ca02
commit 10da456
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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