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
86f8e253612081823a5e1b6944bd02701513cad1784c9dce582c63218d4915ebRUN
|5
09/30/2022 12:55 PM UTC
113d6d8608c6b8e664ec1cd000d497f02536f8bec4257842e5a53370895b240aARG
INSTALL_DISTRIBUTED_EMBEDDINGS=true
09/30/2022 12:54 PM UTC
6ff98a2298baea8aa13bc334b783b0ae5c1c142e36257c6812d90f1714988a1cRUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v4.0 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git if [ "$HUGECTR_DEV_MODE" == "false" ]; then git clone --branch ${HUGECTR_VER} --depth 1 https://${_CI_JOB_TOKEN}${_HUGECTR_REPO} /hugectr &&
  pushd /hugectr &&
  pip install ninja &&
  git submodule update --init --recursive &&
  cd sparse_operation_kit &&
  python setup.py install &&
  cd ../hierarchical_parameter_server &&
  python setup.py install &&
  popd; fi
09/30/2022 12:54 PM UTC
58bbaa5243ca831ea3b55297261c507a77a382c90d8ebe59843c977b88bf3cddRUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v4.0 _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
09/30/2022 12:43 PM UTC
8e48d017a1c753cf6c113a5b96983c5b1f33efdab520eb57a1cff9d16af915d5RUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v4.0 _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
09/30/2022 12:43 PM UTC
de17ed2362b12588430bd659a76684200a9ac0f8167c08526a6e4ad97cb4ad16ARG
HUGECTR_VER=master
09/30/2022 12:43 PM UTC
7fb066cd14e30e0fe5d1ab02468c6d95fe2a2454e694860bd6707ba6472a2a5dARG
_CI_JOB_TOKEN=
09/30/2022 12:43 PM UTC
e39aa1992593a930bc1dbe18a3511bfc22228e3afe2ad5bd52361f29a661ad21ARG
_HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git
09/30/2022 12:43 PM UTC
6fa1ce86b1a375013c09399e67673aecc33df91745a64c74a6b7a7a41d1b0a80ARG
HUGECTR_DEV_MODE=false
09/30/2022 12:43 PM UTC
bfa010afc469788279a5c46e233d314771f37b0fae9ec7ef01c0f516cbe3d39eENV
LD_LIBRARY_PATH=/usr/local/hugectr/lib:/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:/opt/tritonserver/lib LIBRARY_PATH=/usr/local/hugectr/lib:/usr/local/cuda/lib64/stubs: SOK_COMPILE_UNIT_TEST=ON
09/30/2022 12:43 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.