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
12bece841efc92962e706d217c93e69404e761c018cb2080d9baa277dd020704RUN
|10
04/27/2023 11:15 AM UTC
24824a2ac33e7b96e9c77ba3a1a78fffc687e2194e997f090188f3f0cd5f338cRUN
HUGECTR_BACKEND_VER=v23.04.00 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.04.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=23.02 _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
04/27/2023 11:15 AM UTC
a484007e5e337aeb5bbaab6031f5abc84b04b9fd4dd3e46a33037a9c30dd9340RUN
HUGECTR_BACKEND_VER=v23.04.00 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.04.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=23.02 _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
04/27/2023 11:15 AM UTC
3b290d89a4f09e64cd92fd598b383e6fc35e0d45c51e5658ab6452a32c7c096fRUN
HUGECTR_BACKEND_VER=v23.04.00 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.04.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=23.02 _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
04/27/2023 11:15 AM UTC
2260e2f406d35980f5845ebea92c75f0c5933687c41c2ae4be2347cc75adadc3RUN
HUGECTR_BACKEND_VER=v23.04.00 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.04.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=23.02 _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 &&
  rm -rf /repos ; fi
04/27/2023 11:15 AM UTC
0f4cb283161e57e357237ac89eba504f97c03d83f856520982b68c57ac825f0eARG
TRITON_VERSION
04/27/2023 11:15 AM UTC
bec1804a9f8624a18d88aee6760c773cd2c7bfe6f3b1ffdadac42d3cc0356b8aARG
_HUGECTR_BACKEND_REPO=github.com/triton-inference-server/hugectr_backend.git
04/27/2023 11:15 AM UTC
b85dec478c81b254841121d9a3e2c6697773c0588c8d0cbd0600e2da76e34ab0ENV
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::/opt/hadoop/bin:/opt/hadoop/sbin:/usr/local/hugectr/bin:/usr/local/cuda/bin:/usr/lib/x86_64-linux-gnu/:/usr/local/hugectr/bin CPATH=:/usr/local/hugectr/include:/usr/local/hugectr/include LD_LIBRARY_PATH=/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda-11/lib64:/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:/opt/tritonserver/lib:/usr/local/hugectr/lib:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/lib:/repos/dist/lib:/usr/local/hugectr/lib PYTHONPATH=:/usr/local/lib/python3.8/dist-packages/:/usr/local/hugectr/lib
04/27/2023 11:15 AM UTC
0552533b04db209475b91f172dfc5669b5e5cedf161331585da0cd66384513a2RUN
HUGECTR_BACKEND_VER=v23.04.00 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.04.00 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 git clone --branch ${HUGECTR_VER} --depth 1 https://${_CI_JOB_TOKEN}${_HUGECTR_REPO} /hugectr &&
  cd /hugectr &&
  git submodule update --init --recursive &&
  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;90" -DENABLE_MULTINODES=ON .. ; else cmake -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80;90" -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 &&
  mv /hugectr/ci ~/hugectr-ci &&
  rm -rf /hugectr &&
  mkdir -p /hugectr &&
  mv ~/hugectr-ci /hugectr/ci ; fi
04/27/2023 11:15 AM UTC
5c517197fd8a34fa4093379a354baa84a5c1d339025032eb33ec8141b96cd29bARG
HUGECTR_HOME=/usr/local/hugectr
04/27/2023 10:30 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.