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
c1180587726e0d0affe87e17d08bc16902fb1761fb6c33992ed4f2426e73a3c2RUN
|8
01/15/2024 3:42 AM UTC
d33e6b306545a18e212413bdb5f7abb7915b423f1c1ecce634ee152b13c69959RUN
HUGECTR_BACKEND_VER=v23.09.00 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.12.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git rm -rf /usr/local/share/jupyter/lab/staging/node_modules/marked
01/15/2024 3:42 AM UTC
1b43ddf74ca99593935e37f7397e6946e393a770273c74940b88526fb4eeffc2ENV
PYTHONPATH=:/usr/local/lib/python3.10/dist-packages/:/usr/local/hugectr/lib
01/15/2024 3:42 AM UTC
08930832abc3f75396c13851cae96f43c11f62a30e2d5c44c6b900650f0845caRUN
HUGECTR_BACKEND_VER=v23.09.00 HUGECTR_DEV_MODE=false HUGECTR_HOME=/usr/local/hugectr HUGECTR_VER=v23.12.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 ${HUGECTR_HOME}/lib/libgmock* ${HUGECTR_HOME}/lib/pkgconfig/gmock* ${HUGECTR_HOME}/include/gmock &&
  rm -rf ${HUGECTR_HOME}/lib/libgtest* ${HUGECTR_HOME}/lib/pkgconfig/gtest* ${HUGECTR_HOME}/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-$(echo $CUDA_VERSION | awk -F'.' '{print $1"."$2}')/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
01/15/2024 3:42 AM UTC
3c38597c870f692801640734ea824c2be5361210c64111262c27d4035fdfbb19ARG
HUGECTR_HOME=/usr/local/hugectr
01/15/2024 3:25 AM UTC
3bd0d7a74d1b8d0a211609e6df32e88aff46b8ebeea32902c502d0a341a9e405RUN
HUGECTR_BACKEND_VER=v23.09.00 HUGECTR_DEV_MODE=false HUGECTR_VER=v23.12.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git ln -s libibverbs.so.1 $(find /usr/lib/*-linux-gnu/libibverbs.so.1 | sed -e 's/\.1$//g')
01/15/2024 3:25 AM UTC
9d4f1804b17be593a1f9a4d61ebac5def48f6f18b50d1c139360058ac3ebbda0RUN
HUGECTR_BACKEND_VER=v23.09.00 HUGECTR_DEV_MODE=false HUGECTR_VER=v23.12.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git ln -s /usr/lib/libcudf.so /usr/lib/libcudf_io.so
01/15/2024 3:25 AM UTC
8955ec222d3a6fa7f0b50ea2e30ba28db046dd881abbd2ce7f7e059804fe6ca5RUN
HUGECTR_BACKEND_VER=v23.09.00 HUGECTR_DEV_MODE=false HUGECTR_VER=v23.12.00 HWLOC_VER=2.4.1 INSTALL_HDFS=false _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git ln -s /usr/lib/libcudf.so /usr/lib/libcudf_base.so
01/15/2024 3:25 AM UTC
39f0d92b225cc83b2875f0c4c117f886d68447491bc709b2be6ae5e8a347ea5aENV
HCOLL_ENABLE_MCAST=0
01/15/2024 3:25 AM UTC
484d43976eae1e62af12bdee651fbe79c3cbf769475434215d813a5c907df53dENV
SHARP_COLL_LOG_LEVEL=3
01/15/2024 3:25 AM UTC
...