Skip to content
This repository was archived by the owner on Oct 25, 2024. It is now read-only.

Update build.dockerfile #27

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions docker/build.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ LABEL maintainer="[email protected]"
ARG cpu_base_image="tensorflow/build:latest-python3.7"
ARG base_image=$cpu_base_image
ARG tensorflow_pip="tf-nightly"
ARG python_version="python3.7"
ARG python_version="3.7"
ARG APT_COMMAND="apt-get -o Acquire::Retries=3 -y"

# Stops tzdata from asking about timezones and blocking install on user input.
Expand Down Expand Up @@ -73,10 +73,10 @@ ARG pip_dependencies=' \
portpicker'

RUN for python in ${python_version}; do \
$python get-pip.py && \
$python -mpip uninstall -y tensorflow tensorflow-gpu tf-nightly tf-nightly-gpu && \
$python -mpip --no-cache-dir install ${tensorflow_pip} --upgrade && \
$python -mpip --no-cache-dir install $pip_dependencies; \
python${python} get-pip.py && \
python${python} -mpip uninstall -y tensorflow tensorflow-gpu tf-nightly tf-nightly-gpu && \
python${python} -mpip --no-cache-dir install ${tensorflow_pip} --upgrade && \
python${python} -mpip --no-cache-dir install $pip_dependencies; \
done
RUN rm get-pip.py

Expand Down