NVIDIA
NVIDIA
Merlin TensorFlow
Container
NVIDIA
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:33b8a9070a158273ee25d2fb6eb31a2ffd4cc4593a89e3898e76074f7d0e30f5RUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v23.06.00 INSTALL_DISTRIBUTED_EMBEDDINGS=false TFDE_VER=v0.3 _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;
06/23/2023 9:03 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
TFDE_VER=v0.3
06/23/2023 8:47 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
INSTALL_DISTRIBUTED_EMBEDDINGS=false
06/23/2023 8:47 AM UTC
sha256:35a33f5cb9c23f57a450ba23c383f54ba5e7dad775f2810ae5d18be4d5cc24c2RUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v23.06.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
06/23/2023 8:47 AM UTC
sha256:aec2852e47bc96ab8a18d88933692b2b6c9ece1ed81a406e476d8b7ed9082722RUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v23.06.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
06/23/2023 8:47 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
06/23/2023 8:47 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
HUGECTR_VER=main
06/23/2023 8:47 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
_CI_JOB_TOKEN=
06/23/2023 8:47 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
_HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git
06/23/2023 8:47 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
HUGECTR_DEV_MODE=false
06/23/2023 8:47 AM UTC
...