NVIDIA
NVIDIA
Merlin Training
Container
NVIDIA
NVIDIA
Merlin Training

This container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with HugeCTR.

LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CONFIG
032e4238e95afa728f5f1781cba205a1726d9956fb6840e2a8e2eda5d132433cRUN
/bin/bash -c #(nop) ENTRYPOINT []
05/15/2021 7:55 AM UTC
37a77f388e014c14995973113283cfd4fea94b28aaf765af2f0e6180d526f676RUN
/bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
05/15/2021 7:55 AM UTC
f99b6902df5feb07f7567bdd79469cc71edc2c7cfefea3b8537adb8cdbe89c9fRUN
CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.2 RELEASE=true RMM_VER=v0.19.0 /bin/bash -c echo $(du -h --max-depth=1 /)
05/15/2021 7:55 AM UTC
8c48ba2ffaadd0df057838cb25335fa448a901d03c249ff963fc6acc1c99f7d6RUN
CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.2 RELEASE=true RMM_VER=v0.19.0 /bin/bash -c rm -rf /repos
05/15/2021 7:55 AM UTC
faff1b0ebbc787d7eb08a7681797e57c7d9ca3e8a3ab11e2712c000ad4f3a553RUN
/bin/bash -c #(nop) ENV PYTHONPATH=:/usr/local/hugectr/lib
05/15/2021 7:55 AM UTC
a4ad556af7e3165a11ea52b6f63e669fec50d735297452a47e0090f47fa7ee5bRUN
/bin/bash -c #(nop) ENV LD_LIBRARY_PATH=/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/local/hugectr/lib PATH=/usr/local/cuda/lib64/:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/cuda/bin:/usr/lib/x86_64-linux-gnu/:/usr/local/hugectr/bin
05/15/2021 7:55 AM UTC
0fd9ed903442cdff9cbf2ccd302ddd4af00cc730ae896942489dafb782b5ee53RUN
CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.2 RELEASE=true RMM_VER=v0.19.0 /bin/bash -c rm /usr/local/cuda/lib64/stubs/libcuda.so.1
05/15/2021 7:55 AM UTC
dcf3a5146e8f264265d487e77a181d4d573c5f40864ad325c0c80f9d987f681bRUN
CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.2 RELEASE=true RMM_VER=v0.19.0 /bin/bash -c cd /var/tmp/HugeCTR &&
  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/ &&
  cmake -DCMAKE_CXX_COMPILER=/usr/bin/g++ -DCMAKE_C_COMPILER=/usr/bin/gcc -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80" -DVAL_MODE=OFF -DENABLE_MULTINODES=ON -DNCCL_A2A=ON .. &&
  make -j$(nproc) &&
  make install &&
  chmod +x /usr/local/hugectr/bin/* &&
  chmod +x /usr/local/hugectr/lib/* &&
  rm -rf /var/tmp/HugeCTR;
05/15/2021 7:55 AM UTC
11dde8f082bdfbcc28ff08afdb1a996f5a966e20f82a48e38cd99098ae89765bRUN
CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.2 RELEASE=true RMM_VER=v0.19.0 /bin/bash -c mkdir -p /var/tmp &&
  cd /var/tmp &&
  git clone https://github.com/NVIDIA/HugeCTR.git HugeCTR &&
  cd - &&
  cd /var/tmp/HugeCTR &&
  if [ "$RELEASE" == "true" ] &&
  [ ${HUGECTR_VER} != 0 ]; then git fetch --all --tags &&
  git checkout tags/${HUGECTR_VER}; else git checkout master; fi &&
  git submodule update --init --recursive
05/15/2021 7:46 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.