Skip to content

Temporary default to CPU container #14

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 3 commits into from
Closed

Temporary default to CPU container #14

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
745d4a9
Temporary default mode CPU
bhack Jun 5, 2020
75e373f
Wrong commited file
bhack Jun 5, 2020
fcad0f8
Fix duplicated Tensorflow wheel layers
bhack Jun 5, 2020
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
13 changes: 7 additions & 6 deletions tools/docker/build_wheel.Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,9 +1,13 @@
#syntax=docker/dockerfile:1.1.5-experimental
ARG TF_VERSION
ARG PY_VERSION
FROM tfaddons/tensorflow:2.1.0-custom-op-gpu-ubuntu16-minimal as base_install
ENV TF_NEED_CUDA="1"
ARG IMAGE_TYPE
ARG TF_VERSION
FROM tensorflow/tensorflow:$TF_VERSION-custom-op$IMAGE_TYPE-ubuntu16 as base_install

ARG TF_NEED_CUDA="0"
ENV TF_NEED_CUDA=$TF_NEED_CUDA

# TODO: This ENV block is useless when we build the CPU IMAGE
# is needed because when we sqashed the image, we lost all environment variables.
ENV NVIDIA_REQUIRE_CUDA=cuda>=10.1 brand=tesla,driver>=384,driver<385 brand=tesla,driver>=396,driver<397 brand=tesla,driver>=410,driver<411
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
Expand All @@ -26,9 +30,6 @@ RUN ln -sf $(which python$PY_VERSION) /usr/bin/python3

RUN python -m pip install --upgrade pip==19.0 auditwheel==2.0.0

ARG TF_VERSION
RUN python -m pip install --default-timeout=1000 tensorflow==$TF_VERSION

COPY tools/install_deps/ /install_deps
RUN python -m pip install -r /install_deps/pytest.txt

Expand Down