From fe0c98d82abc222512eb7b2cd848cec1b5d982e2 Mon Sep 17 00:00:00 2001 From: Chandra Devarakonda Date: Wed, 13 Mar 2024 18:46:40 +0000 Subject: [PATCH] Minor fix in the packages --- tensorflow_runtime_dockerfiles/tpu.Dockerfile | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/tensorflow_runtime_dockerfiles/tpu.Dockerfile b/tensorflow_runtime_dockerfiles/tpu.Dockerfile index a69400d..73d88e9 100644 --- a/tensorflow_runtime_dockerfiles/tpu.Dockerfile +++ b/tensorflow_runtime_dockerfiles/tpu.Dockerfile @@ -19,9 +19,9 @@ ENV LANG C.UTF-8 COPY setup.sources.sh /setup.sources.sh COPY setup.packages.sh /setup.packages.sh -COPY cpu.packages.txt /cpu.packages.txt +COPY tpu.packages.txt /tpu.packages.txt RUN /setup.sources.sh -RUN /setup.packages.sh /cpu.packages.txt +RUN /setup.packages.sh /tpu.packages.txt ARG PYTHON_VERSION=python3.11 @@ -30,15 +30,6 @@ COPY setup.python.sh /setup.python.sh COPY cpu.requirements.txt /tpu.requirements.txt RUN /setup.python.sh $PYTHON_VERSION /tpu.requirements.txt RUN pip install --no-cache-dir ${TENSORFLOW_PACKAGE} -f https://storage.googleapis.com/libtpu-tf-releases/index.html -ENV NEXT_PLUGGABLE_DEVICE_USE_C_API=true -ENV TF_PLUGGABLE_DEVICE_LIBRARY_PATH="" # Initialize - -# Dynamically determine libtpu installation path -RUN python -c "import libtpu; import os; print(os.path.join(os.path.dirname(libtpu.__file__), 'libtpu.so'))" \ - | tee /out/.libtpu_path # Capture the output - -# Set the environment variable -ENV TF_PLUGGABLE_DEVICE_LIBRARY_PATH=$(cat /out/.libtpu_path) COPY bashrc /etc/bash.bashrc RUN chmod a+rwx /etc/bash.bashrc