Skip to main content
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
    Merlin Tensorflow Training
    Container
    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"]
    		08/04/2021 6:35 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENTRYPOINT []
    		08/04/2021 6:35 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
    		08/04/2021 6:35 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 /bin/bash -c echo $(du -h --max-depth=1 /)
    		08/04/2021 6:35 PM UTC
    sha256:c78e402928e0f25a78264c40b4e08d43e17fde1e2c1aed82aa259654647d7398RUN
    CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 /bin/bash -c pip uninstall pandas -y; pip install pandas==1.1.5
    		08/04/2021 6:34 PM UTC
    sha256:1219046e661729ff34047bcc364b008e40a45b4e852822d80dd95459e0e49c3fRUN
    CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 /bin/bash -c pip install dask[dataframe]==2021.04.1
    		08/04/2021 6:34 PM UTC
    sha256:ad5e2ef0fe55d3988a9588a4f85dddad07bb759f6bb32dc2cacbde3af98b2142RUN
    CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 /bin/bash -c pip install dask==2021.04.1 distributed==2021.04.1 dask-cuda
    		08/04/2021 6:34 PM UTC
    sha256:a92c620b89608f29ae1499b74ef9ba736e871f43153b0c14f0feb145e4bd8540RUN
    CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 /bin/bash -c pip uninstall numpy -y; pip install numpy
    		08/04/2021 6:34 PM UTC
    sha256:85043d814bd451423347b22444056e0fab12068759a90154f78facd6412446d8RUN
    CUDF_VER=v21.06.01 HUGECTR_VER=v3.1 NVTAB_VER=v0.6.0 RELEASE=true RMM_VER=v21.06.00 SM=60;61;70;75;80 /bin/bash -c git clone https://github.com/rapidsai/asvdb.git build-env &&
  pushd build-env &&
  python setup.py install &&
  popd &&
  rm -rf build-env
    		08/04/2021 6:33 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENV LD_LIBRARY_PATH=/usr/local/hugectr/lib:/usr/local/cuda/extras/CUPTI/lib64:/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 LIBRARY_PATH=/usr/local/hugectr/lib:/usr/local/cuda/lib64/stubs: PYTHONPATH=/usr/local/hugectr/lib:
    		08/04/2021 6:33 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.