NVIDIA
Merlin TensorFlow
Container
NVIDIA
Merlin TensorFlow

The Merlin TensorFlow container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with TensorFlow, and serve the trained model on Triton Inference Server.

  • LayerLabelCreated
    sha256:3b264e5294f1c9051d5522ac867972c846edda5365714680ed5372e2e9cd8547RUN
    HUGECTR_DEV_MODE=false HUGECTR_VER=v23.02.00 INSTALL_DISTRIBUTED_EMBEDDINGS=true TFDE_VER=v0.2 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git if [ "$HUGECTR_DEV_MODE" == "false" ]; then git clone --branch ${HUGECTR_VER} --depth 1 --recurse-submodules --shallow-submodules https://${_CI_JOB_TOKEN}${_HUGECTR_REPO} /hugectr &&
      pushd /hugectr &&
      rm -rf .git/modules &&
      pip --no-cache-dir install ninja tf2onnx &&
      cd sparse_operation_kit &&
      python setup.py install &&
      cd ../hps_tf &&
      python setup.py install &&
      popd &&
      mv /hugectr/ci ~/hugectr-ci &&
      mv /hugectr/sparse_operation_kit ~/hugectr-sparse_operation_kit &&
      rm -rf /hugectr &&
      mkdir -p /hugectr &&
      mv ~/hugectr-ci /hugectr/ci &&
      mv ~/hugectr-sparse_operation_kit /hugectr/sparse_operation_kit; fi; if [ "$INSTALL_DISTRIBUTED_EMBEDDINGS" == "true" ]; then git clone --branch ${TFDE_VER} --depth 1 https://github.com/NVIDIA-Merlin/distributed-embeddings.git /distributed_embeddings/ &&
      cd /distributed_embeddings &&
      git submodule update --init --recursive &&
      make pip_pkg &&
      pip install --no-cache-dir artifacts/*.whl &&
      make clean; fi;
    03/09/2023 9:59 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TFDE_VER=v0.2
    03/09/2023 9:44 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    INSTALL_DISTRIBUTED_EMBEDDINGS=true
    03/09/2023 9:44 AM UTC
    sha256:5f1fc79c7204c7bdb8542ac8f1a5fc86389fd495da9da040bce448ec6f1f3286RUN
    HUGECTR_DEV_MODE=false HUGECTR_VER=v23.02.00 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git ln -s /usr/lib/x86_64-linux-gnu/libibverbs.so.1 /usr/lib/x86_64-linux-gnu/libibverbs.so
    03/09/2023 9:44 AM UTC
    sha256:fbefdae4f256287f67ae5f2c344959e1fa4a9fb29f28632f26f27f022bf91ae0RUN
    HUGECTR_DEV_MODE=false HUGECTR_VER=v23.02.00 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git mkdir -p /usr/local/nvidia/lib64 &&
      ln -s /usr/local/cuda/lib64/libcusolver.so /usr/local/nvidia/lib64/libcusolver.so.10
    03/09/2023 9:44 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    CPATH=:/usr/local/hugectr/include:/include LD_LIBRARY_PATH=/lib:/usr/local/lib/python3.8/dist-packages/tensorflow:/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 LIBRARY_PATH=/lib:/usr/local/cuda/lib64/stubs: SOK_COMPILE_UNIT_TEST=ON
    03/09/2023 9:44 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    HUGECTR_VER=main
    03/09/2023 9:44 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    _CI_JOB_TOKEN=
    03/09/2023 9:44 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git
    03/09/2023 9:44 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    HUGECTR_DEV_MODE=false
    03/09/2023 9:44 AM UTC
    ...