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