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 PyTorch Training
    Container
    NVIDIA
    Merlin PyTorch Training

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

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) CMD ["/bin/bash"]
    		09/27/2021 5:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENTRYPOINT []
    		09/27/2021 5:26 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
    		09/27/2021 5:26 PM UTC
    sha256:ce58b2ba32479b8d8582dc8afcd018bae708526f52526b9ea9401b3fa8552addRUN
    CUDF_VER=v21.08.02 NVTAB_VER=v0.7.0 RELEASE=True RMM_VER=v21.08.02 /bin/bash -c echo $(du -h --max-depth=1 /)
    		09/27/2021 5:26 PM UTC
    sha256:a63e2bf0879592329fe9dfcd8738af0f377a9e357e6692ce536d456623f9c4fdRUN
    CUDF_VER=v21.08.02 NVTAB_VER=v0.7.0 RELEASE=True RMM_VER=v21.08.02 /bin/bash -c rm -rf /repos
    		09/27/2021 5:26 PM UTC
    sha256:fe418e28171aee5e5bd1dee094f665d4ea829c471997c4fedc0a832d6febcb91RUN
    CUDF_VER=v21.08.02 NVTAB_VER=v0.7.0 RELEASE=True RMM_VER=v21.08.02 /bin/bash -c pip install torchmetrics==0.3.2
    		09/27/2021 5:26 PM UTC
    sha256:37a1b341506208f74ad321fdc819bb87cafb8e2f69671322ea8d643344e8ce76RUN
    CUDF_VER=v21.08.02 NVTAB_VER=v0.7.0 RELEASE=True RMM_VER=v21.08.02 /bin/bash -c pip install dask==2021.07.1 distributed==2021.07.1 dask[dataframe]==2021.07.1
    		09/27/2021 5:25 PM UTC
    sha256:164abd9815f27222c83c7c10a9ec6aa1665deb1e5a24ed263abbc3db3c70b5cfRUN
    CUDF_VER=v21.08.02 NVTAB_VER=v0.7.0 RELEASE=True RMM_VER=v21.08.02 /bin/bash -c pip install nvtx pandas==1.1.5 mpi4py==3.0.3 cupy-cuda113 cachetools typing_extensions fastavro
    		09/27/2021 5:25 PM UTC
    sha256:f8ed869ea00cd417773d3407f7edc8667a635489d7e0c36ff63caed1585408b3RUN
    CUDF_VER=v21.08.02 NVTAB_VER=v0.7.0 RELEASE=True RMM_VER=v21.08.02 /bin/bash -c CC=/usr/bin/gcc CXX=/usr/bin/g++ HOROVOD_CUDA_HOME=/usr/local/cuda/ HOROVOD_BUILD_CUDA_CC_LIST=60,70,75,80 HOROVOD_GPU_OPERATIONS=NCCL HOROVOD_WITH_PYTORCH=1 HOROVOD_NCCL_LINK=SHARED pip install --no-cache-dir horovod[pytorch]
    		09/27/2021 5:24 PM UTC
    sha256:ba9695bf650f0ceb0797ef3416ea031f57f73edff669da77353da246c047274bRUN
    CUDF_VER=v21.08.02 NVTAB_VER=v0.7.0 RELEASE=True RMM_VER=v21.08.02 /bin/bash -c pip install fastai
    		09/27/2021 5:19 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.