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
ff32d61f8ef36dd73da620de671016591d28c7ba9a0dba218897ff77a39bd90bRUN
|17
08/01/2022 11:23 AM UTC
823cead1cc7c9eeadaca6d80b4e01d82948dabb1fc699d0192dea976e0da6a62RUN
BUILD_HADOOP=false HADOOP_VER=3.3.2 HIREDIS_VER=1.0.2 HUGECTR_BACKEND_VER=v3.7 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v3.8 HWLOC_VER=2.4.1 INSTALL_HDFS=false PROTOC_VER=3.19.4 RDKAFKA_VER=1.8.2 REDIS_PP_VER=1.3.3 ROCKSDB_VER=6.29.3 TRITON_VERSION=22.06 _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
08/01/2022 11:23 AM UTC
9ef2a16c83be5a6ac00101372f861e8fc2bddc056c0e5ae1b07014e8842d419dRUN
BUILD_HADOOP=false HADOOP_VER=3.3.2 HIREDIS_VER=1.0.2 HUGECTR_BACKEND_VER=v3.7 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v3.8 HWLOC_VER=2.4.1 INSTALL_HDFS=false PROTOC_VER=3.19.4 RDKAFKA_VER=1.8.2 REDIS_PP_VER=1.3.3 ROCKSDB_VER=6.29.3 TRITON_VERSION=22.06 _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
08/01/2022 11:23 AM UTC
a3629746d223d3af8b3eb4158b7c22a49036bfc771a93e9b100bb6e7e0a498c8RUN
BUILD_HADOOP=false HADOOP_VER=3.3.2 HIREDIS_VER=1.0.2 HUGECTR_BACKEND_VER=v3.7 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v3.8 HWLOC_VER=2.4.1 INSTALL_HDFS=false PROTOC_VER=3.19.4 RDKAFKA_VER=1.8.2 REDIS_PP_VER=1.3.3 ROCKSDB_VER=6.29.3 TRITON_VERSION=22.06 _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
08/01/2022 11:23 AM UTC
bf579a4e9064a7061c14f90061db160de23b4c4ba08c26dac6db1e7eaf8229baRUN
BUILD_HADOOP=false HADOOP_VER=3.3.2 HIREDIS_VER=1.0.2 HUGECTR_BACKEND_VER=v3.7 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v3.8 HWLOC_VER=2.4.1 INSTALL_HDFS=false PROTOC_VER=3.19.4 RDKAFKA_VER=1.8.2 REDIS_PP_VER=1.3.3 ROCKSDB_VER=6.29.3 TRITON_VERSION=22.06 _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
08/01/2022 11:23 AM UTC
fd7aac4d782d4034c025fe0e5d1156852f2257e8c879cece4aee096ee07a4579RUN
BUILD_HADOOP=false HADOOP_VER=3.3.2 HIREDIS_VER=1.0.2 HUGECTR_BACKEND_VER=v3.7 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v3.8 HWLOC_VER=2.4.1 INSTALL_HDFS=false PROTOC_VER=3.19.4 RDKAFKA_VER=1.8.2 REDIS_PP_VER=1.3.3 ROCKSDB_VER=6.29.3 TRITON_VERSION=22.06 _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
08/01/2022 11:22 AM UTC
3d9e8f90d5e135001c599a95c5c4457380dd216a0bd00840c7b2454e62a116d1ARG
TRITON_VERSION
08/01/2022 11:22 AM UTC
f5d65464f6380f87eb05dee621056dd547b5cfb97ea8a7552cc3dce0f1d46970ARG
_HUGECTR_BACKEND_REPO=github.com/triton-inference-server/hugectr_backend.git
08/01/2022 11:22 AM UTC
4700b11cd30da4ca42d5d95f3f4067d6c0eafceb50869454bff3ac88012e0b3bENV
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
08/01/2022 11:22 AM UTC
3c68162047066debb0c6051a3b9f77d1717642a33c00c0de4e1b1a9e4df59516RUN
BUILD_HADOOP=false HADOOP_VER=3.3.2 HIREDIS_VER=1.0.2 HUGECTR_BACKEND_VER=v3.7 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v3.8 HWLOC_VER=2.4.1 INSTALL_HDFS=false PROTOC_VER=3.19.4 RDKAFKA_VER=1.8.2 REDIS_PP_VER=1.3.3 ROCKSDB_VER=6.29.3 _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 [[ -f "/usr/local/lib/libhdfs.so" ]]; then cmake -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80" -DENABLE_MULTINODES=ON -DENABLE_HDFS=ON .. ; else cmake -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80" -DENABLE_MULTINODES=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
08/01/2022 11:22 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.