Skip to content
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

add Tf keras to dockerfile #341

Merged
merged 19 commits into from
May 27, 2022
Merged

add Tf keras to dockerfile #341

Changes from all commits
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
cb2341a
add entrypoint to all containers
jperez999 May 2, 2022
e217d5d
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 3, 2022
59fa46e
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 3, 2022
41bfd49
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 4, 2022
fa45e64
remove -e for pip installs
jperez999 May 5, 2022
f972188
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 5, 2022
e09c1d0
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 9, 2022
98844f7
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 11, 2022
e74d5c1
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 13, 2022
0b30e36
add keyring to fix bad key nvidia issue
jperez999 May 13, 2022
20bf210
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 18, 2022
57bbeea
trying to move env call to see if it helps fix the install of distrib…
jperez999 May 18, 2022
434c1bb
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 21, 2022
930361d
fix extra CVE
jperez999 May 21, 2022
863ea68
Merge branch 'main' of https://github.com/NVIDIA-Merlin/Merlin
jperez999 May 24, 2022
9a32725
fixing CI routes
jperez999 May 24, 2022
1369bcd
merge in resolved
jperez999 May 27, 2022
96576b6
add keras to pull from upstream container
jperez999 May 27, 2022
0fc45a2
fix for tf install in container
jperez999 May 27, 2022
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
12 changes: 7 additions & 5 deletions docker/dockerfile.tf
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
# syntax=docker/dockerfile:1.2
ARG MERLIN_VERSION=22.05-pre
ARG TRITON_VERSION=22.03
ARG TENSORFLOW_VERSION=22.03
ARG MERLIN_VERSION=22.06
ARG TRITON_VERSION=22.05
ARG TENSORFLOW_VERSION=22.05

ARG DLFW_IMAGE=nvcr.io/nvidia/tensorflow:${TENSORFLOW_VERSION}-tf2-py3
ARG FULL_IMAGE=nvcr.io/nvidia/tritonserver:${TRITON_VERSION}-py3
ARG BASE_IMAGE=merlin/base:${MERLIN_VERSION}
ARG BASE_IMAGE=nvcr.io/nvstaging/merlin/merlin-base:${MERLIN_VERSION}

FROM ${DLFW_IMAGE} as dlfw
FROM ${FULL_IMAGE} as triton
Expand All @@ -16,13 +16,15 @@ COPY --chown=1000:1000 --from=triton /opt/tritonserver/backends/tensorflow2 back

# Tensorflow dependencies (only)
RUN pip install tensorflow-gpu \
&& rm -r /usr/local/lib/python3.8/dist-packages/tensorflow
&& pip uninstall tensorflow-gpu keras -y

# DLFW Tensorflow packages
COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python3.8/dist-packages/tensorflow /usr/local/lib/python3.8/dist-packages/tensorflow/
COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python3.8/dist-packages/keras /usr/local/lib/python3.8/dist-packages/keras/
COPY --chown=1000:1000 --from=dlfw /usr/local/lib/tensorflow/ /usr/local/lib/tensorflow/
COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python3.8/dist-packages/horovod /usr/local/lib/python3.8/dist-packages/horovod/
COPY --chown=1000:1000 --from=dlfw /usr/local/bin/horovodrun /usr/local/bin/horovodrun


HEALTHCHECK NONE
CMD ["/bin/bash"]