ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
NVIDIA
NVIDIA
Merlin Tensorflow Training
Container
NVIDIA
NVIDIA
Merlin Tensorflow Training

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

LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
/bin/bash -c #(nop) CMD ["/bin/bash"]
09/25/2021 9:32 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
/bin/bash -c #(nop) ENTRYPOINT []
09/25/2021 9:32 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
/bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
09/25/2021 9:32 PM UTC
sha256:6536d4263d7590248ac2659e545350570aa55871d2d7b36bd30711089b77504eRUN
HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c echo $(du -h --max-depth=1 /)
09/25/2021 9:32 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
/bin/bash -c #(nop) SHELL [/bin/bash -c]
09/25/2021 9:32 PM UTC
sha256:140de424640898b395be478a1c7962f72c6d0962f03574bb94ccbd007d04e92fRUN
HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c pip install numba numpy --upgrade
09/25/2021 9:32 PM UTC
sha256:db9061fee435594828666bdc9a728bccfac54f281878d3cbdc663ef39f6f342dRUN
HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c pip install pybind11
09/25/2021 9:32 PM UTC
sha256:14630817fb063a981a8ea66e47f81c568df0d4b9c2f91ca3ed2a4dece2f75506RUN
HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c git clone https://github.com/NVIDIA/HugeCTR.git build-env &&
  pushd build-env &&
  if [ "$RELEASE" == "true" ] &&
  [ ${HUGECTR_VER} != "vnightly" ] ; then git fetch --all --tags &&
  git checkout tags/${HUGECTR_VER}; else echo ${HUGECTR_VER} &&
  git checkout ${HUGECTR_VER}-integration; fi &&
  cd sparse_operation_kit &&
  bash ./install.sh --SM=$SM --USE_NVTX=$USE_NVTX &&
  popd &&
  rm -rf build-env &&
  rm -rf /var/tmp/HugeCTR
09/25/2021 9:32 PM UTC
sha256:d4c089d34c2623537082b578e376ce9dfb7916ce1fc59089fc2ab61d2c6d91d4RUN
HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c ln -s /usr/lib/x86_64-linux-gnu/libibverbs.so.1.11.32.1 /usr/lib/x86_64-linux-gnu/libibverbs.so
09/25/2021 9:30 PM UTC
sha256:fda42af4eb407907ce7f916964df5ac0d3b7a047ad4e9219ad9106d4524aae33RUN
HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c mkdir -p /usr/local/nvidia/lib64 &&
  ln -s /usr/local/cuda/lib64/libcusolver.so /usr/local/nvidia/lib64/libcusolver.so.10
09/25/2021 9:30 PM 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.