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
b1bacbbed10306083d57ce09f75d9909519b4c326bbfc2769bfa5e975541f0f0RUN
|10
06/23/2023 9:18 AM UTC
235e4e61931d57daf91e0c519203ef0da27f0d34a89c8bcf299aeee7393a20cdRUN
HUGECTR_BACKEND_VER=release-23.06 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.06.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=23.04 _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
06/23/2023 9:18 AM UTC
bedc08290f8e8c1cfe1f79b22f3624ec9ce1aa787f003b9160df5dc402c55f1eRUN
HUGECTR_BACKEND_VER=release-23.06 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.06.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=23.04 _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
06/23/2023 9:18 AM UTC
e5179218cc5fb4732de642b53043e5b32bb5e592b43170c5ed863475f0f376c7RUN
HUGECTR_BACKEND_VER=release-23.06 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.06.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=23.04 _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
06/23/2023 9:18 AM UTC
6ca8d9b824a2d528f3d06203797a1c7108619f707dc48cf2477d3e0afffcf312RUN
HUGECTR_BACKEND_VER=release-23.06 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.06.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false TRITON_VERSION=23.04 _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
06/23/2023 9:18 AM UTC
0cca5bf56cfb07435025e2f2378d33a821284b128bfe8e6155ff644ea52b7d4bARG
TRITON_VERSION
06/23/2023 9:17 AM UTC
12c352a99197b5155a7d2ea8e885900d2073fc7094651b3d2a076888938e3bfeARG
_HUGECTR_BACKEND_REPO=github.com/triton-inference-server/hugectr_backend.git
06/23/2023 9:17 AM UTC
2f359a20a20d9e27156c5cd139a4df4a0d665b46bc200a1ab3ef097d62347600ENV
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/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
06/23/2023 9:17 AM UTC
6b1802c3a689520b86de804e10b7d9033c3e6053827742d92e16dfa4445a7fd7RUN
HUGECTR_BACKEND_VER=release-23.06 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.06.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 rm -rf /usr/local/hugectr/lib/libgmock* /usr/local/hugectr/lib/pkgconfig/gmock* /usr/local/hugectr/include/gmock &&
  rm -rf /usr/local/hugectr/lib/libgtest* /usr/local/hugectr/lib/pkgconfig/gtest* /usr/local/hugectr/include/gtest &&
  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
06/23/2023 9:17 AM UTC
96d2c717df8252203704e52a3ac1bd223c58fd202f6841b054db12b1ae5d770cARG
HUGECTR_HOME=/usr/local/hugectr
06/23/2023 8:43 AM UTC
...