Skip to content

Commit

Permalink
Minor fix in the packages
Browse files Browse the repository at this point in the history
  • Loading branch information
chandrasekhard2 committed Mar 13, 2024
1 parent 6ba2ba6 commit fe0c98d
Showing 1 changed file with 2 additions and 11 deletions.
13 changes: 2 additions & 11 deletions tensorflow_runtime_dockerfiles/tpu.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down

0 comments on commit fe0c98d

Please sign in to comment.