NVIDIA
NVIDIA
Merlin Inference
Container
NVIDIA
NVIDIA
Merlin Inference

This container allows users to deploy NVTabular workflows and HugeCTR or TensorFlow models to Triton Inference server for production.

LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
/bin/bash -c #(nop) CMD ["/bin/bash"]
08/04/2021 9:33 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
/bin/bash -c #(nop) ENTRYPOINT []
08/04/2021 9:33 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
/bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
08/04/2021 9:33 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 TRITON_VERSION=21.07 /bin/bash -c echo $(du -h --max-depth=1 /)
08/04/2021 9:33 PM UTC
sha256:1a3d226979d2397b3346f4b480883353aef6113be97b522510a8e86cf97d22b3RUN
CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 TRITON_VERSION=21.07 /bin/bash -c pip install tqdm
08/04/2021 9:33 PM UTC
sha256:f4bc77e62a43b6e31c7b112eaad8a4978cf95250b7234a6b8f143a7864f0bf58RUN
CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 TRITON_VERSION=21.07 /bin/bash -c rm -rf /repos
08/04/2021 9:33 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 TRITON_VERSION=21.07 /bin/bash -c echo $(du -h --max-depth=1 /)
08/04/2021 9:33 PM UTC
sha256:567324e674a2d19a64dc65f7d1875ffa982a7ada3572223efea8dfc065f70c69RUN
CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 TRITON_VERSION=21.07 /bin/bash -c ln -s /usr/local/hugectr/backends/hugectr /opt/tritonserver/backends/
08/04/2021 9:33 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
/bin/bash -c #(nop) ENV LD_LIBRARY_PATH=/usr/local/hugectr/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/lib:/repos/dist/lib LIBRARY_PATH=/usr/local/hugectr/lib:/usr/local/cuda/lib64/stubs: PYTHONPATH=/usr/local/hugectr/lib:
08/04/2021 9:33 PM UTC
sha256:19e0aeb4fa4532132618f94b35f7d1f28e5d504118749c59cbffc9743de9559aRUN
CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 TRITON_VERSION=21.07 /bin/bash -c git clone https://github.com/NVIDIA/HugeCTR.git /repos/HugeCTR &&
  cd /repos/HugeCTR &&
  if [ "$RELEASE" == "true" ] &&
  [ ${HUGECTR_VER} != "vnightly" ]; then git fetch --all --tags &&
  git checkout tags/${HUGECTR_VER}; else git checkout master; fi &&
  git submodule update --init --recursive &&
  mkdir -p build &&
  cd build &&
  cmake -DCMAKE_BUILD_TYPE=Release -DSM=$SM -DENABLE_INFERENCE=ON .. &&
  make -j$(nproc) &&
  make install &&
  export CPATH=/usr/local/hugectr/include:$CPATH &&
  export LIBRARY_PATH=/usr/local/hugectr/lib:$LIBRARY_PATH &&
  git clone https://github.com/triton-inference-server/hugectr_backend /repos/hugectr_inference_backend &&
  cd /repos/hugectr_inference_backend &&
  if [ "$RELEASE" == "true" ] &&
  [ ${HUGECTR_VER} != "vnightly" ] ; then git fetch --all --tags &&
  git checkout tags/${HUGECTR_VER}; else git checkout main; fi &&
  mkdir -p build &&
  cd build &&
  cmake -DCMAKE_INSTALL_PREFIX:PATH=/usr/local/hugectr .. -D TRITON_COMMON_REPO_TAG="r$TRITON_VERSION" -D TRITON_CORE_REPO_TAG="r$TRITON_VERSION" -D TRITON_BACKEND_REPO_TAG="r$TRITON_VERSION" &&
  make -j$(nproc) &&
  make install
08/04/2021 9:33 PM UTC
...