NVIDIA
NVIDIA
Merlin HugeCTR
Container
NVIDIA
NVIDIA
Merlin HugeCTR

The Merlin HugeCTR container enables you to perform data preprocessing, feature engineering, train models with HugeCTR, and then serve the trained model with Triton Inference Server.

LayerLabelCreated
9d9996f093f3d32fb2cce73a648a4e51c40842b307cff2ba21e499f370db1befRUN
|10
12/31/2022 2:38 AM UTC
952d9ea428d80f8652f450be89ab12e0523fa8dd6e206b1421e9144de6cb2c70RUN
HUGECTR_BACKEND_VER=v4.3 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v4.3 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=22.11 _CI_JOB_TOKEN= _HUGECTR_BACKEND_REPO=github.com/triton-inference-server/hugectr_backend.git _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git rm -rf /usr/local/share/jupyter/lab/staging/node_modules/marked
12/31/2022 2:38 AM UTC
42fca574047c5f333b9429242fd35ec355f930109acf4906f4a6e03583ce73b9RUN
HUGECTR_BACKEND_VER=v4.3 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v4.3 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=22.11 _CI_JOB_TOKEN= _HUGECTR_BACKEND_REPO=github.com/triton-inference-server/hugectr_backend.git _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git rm -rf /repos
12/31/2022 2:38 AM UTC
effe557e2fd27db90846d84e025c12236bf2a1eaeb5aae988ad7ae42df5af001RUN
HUGECTR_BACKEND_VER=v4.3 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v4.3 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=22.11 _CI_JOB_TOKEN= _HUGECTR_BACKEND_REPO=github.com/triton-inference-server/hugectr_backend.git _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git rm /usr/local/cuda/lib64/stubs/libcuda.so.1
12/31/2022 2:38 AM UTC
8cc43b5d0a986f2a3a9cc447eee44e63ab7eb1663d2430020a52b35897c6e986RUN
HUGECTR_BACKEND_VER=v4.3 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v4.3 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=22.11 _CI_JOB_TOKEN= _HUGECTR_BACKEND_REPO=github.com/triton-inference-server/hugectr_backend.git _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git ln -s ${HUGECTR_HOME}/backends/hugectr /opt/tritonserver/backends/hugectr
12/31/2022 2:38 AM UTC
95c486fcf1a9a32ba468059b0648f3d32b3b90c409aff4258caae55cef38e6b5RUN
HUGECTR_BACKEND_VER=v4.3 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v4.3 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=22.11 _CI_JOB_TOKEN= _HUGECTR_BACKEND_REPO=github.com/triton-inference-server/hugectr_backend.git _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git if [ "${HUGECTR_DEV_MODE}" == "false" ]; then git clone --branch ${HUGECTR_BACKEND_VER} --depth 1 https://${_CI_JOB_TOKEN}${_HUGECTR_BACKEND_REPO} /repos/hugectr_triton_backend &&
  mkdir /repos/hugectr_triton_backend/build &&
  cd /repos/hugectr_triton_backend/build &&
  cmake -DCMAKE_INSTALL_PREFIX:PATH=${HUGECTR_HOME} -DTRITON_COMMON_REPO_TAG="r${TRITON_VERSION}" -DTRITON_CORE_REPO_TAG="r${TRITON_VERSION}" -DTRITON_BACKEND_REPO_TAG="r${TRITON_VERSION}" .. &&
  make -j$(nproc) &&
  make install &&
  cd ../.. &&
  rm -rf hugectr_triton_backend &&
  chmod +x ${HUGECTR_HOME}/lib/*.so ${HUGECTR_HOME}/backends/hugectr/*.so ; fi
12/31/2022 2:38 AM UTC
aa952dea98dd852ba13d52b9d73273e5f5c624d7e02dfc86a5892a70e663c777ARG
TRITON_VERSION
12/31/2022 2:37 AM UTC
2ef8a20e7a80cdfff0e6087beb3d0993279360ee7392160750c7c63487ecb2d3ARG
_HUGECTR_BACKEND_REPO=github.com/triton-inference-server/hugectr_backend.git
12/31/2022 2:37 AM UTC
aeeda3cf2879b15af3408d15110c1659f0c89908840edfc7e20d81271af35c94ENV
PATH=/usr/local/cuda/lib64/:/opt/tritonserver/bin:/usr/local/cuda/lib64/:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/usr/local/cuda/bin::/usr/local/cuda/bin:/usr/lib/x86_64-linux-gnu/:/opt/hadoop/bin:/opt/hadoop/sbin:/usr/local/hugectr/bin CPATH=:/usr/local/hugectr/include LD_LIBRARY_PATH=/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:/opt/tritonserver/lib:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/lib:/repos/dist/lib:/usr/lib/jvm/default-java/lib:/usr/lib/jvm/default-java/lib/server:/usr/local/hugectr/lib PYTHONPATH=:/usr/local/lib/python3.8/dist-packages/:/usr/local/hugectr/lib
12/31/2022 2:37 AM UTC
078de39604c4192b9b94d870e9fe3a7e3a75b29c032732a4ab1a7a48f0affd3eRUN
HUGECTR_BACKEND_VER=v4.3 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v4.3 HWLOC_VER=2.4.1 INSTALL_HDFS=false _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git if [[ "${HUGECTR_DEV_MODE}" == "false" ]]; then cd /hugectr &&
  git submodule update --init --recursive &&
  rm -rf build &&
  mkdir build &&
  cd build &&
  LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH &&
  export PATH=$PATH:/usr/local/cuda-${CUDA_SHORT_VERSION}/compat &&
  if [[ "${INSTALL_HDFS}" == "false" ]]; then cmake -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80" -DENABLE_MULTINODES=ON .. ; else cmake -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80" -DENABLE_MULTINODES=ON -DENABLE_HDFS=ON .. ; fi &&
  make -j$(nproc) &&
  make install &&
  rm -rf ./* &&
  chmod +x ${HUGECTR_HOME}/bin/* ${HUGECTR_HOME}/lib/*.so &&
  cd ../onnx_converter &&
  python setup.py install ; fi
12/31/2022 2:37 AM UTC
...

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.