Skip to content

Commit

Permalink
fix building without pip package
Browse files Browse the repository at this point in the history
  • Loading branch information
lreiher committed Aug 15, 2023
1 parent 52c5d6a commit e2296bd
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,6 @@ RUN if [ "${GPU_POSTFIX}" = "-gpu" ]; then \

# move libtensorflow_cc to separate folder for easier Dockerfile COPY
RUN mkdir -p bazel-bin/tensorflow/lib && \
mkdir -p bazel-bin/tensorflow/pip && \
mv bazel-bin/tensorflow/libtensorflow_cc.so* bazel-bin/tensorflow/lib/ && \
mv bazel-bin/tensorflow/libtensorflow_framework.so* bazel-bin/tensorflow/lib/ && \
ln -sf libtensorflow_framework.so.2 bazel-bin/tensorflow/lib/libtensorflow_framework.so && \
Expand All @@ -104,6 +103,9 @@ RUN if [ "${BUILD_PIP_PACKAGE}" = "1" ]; then \
fi; \
pip install patchelf && \
./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow/pip ; \
else \
mkdir -p /tmp/tensorflow/pip && \
touch /tmp/tensorflow/pip/tensorflow-dummy.whl ; \
fi

# build protobuf from source, same version as TensorFlow is using
Expand Down

0 comments on commit e2296bd

Please sign in to comment.