Skip to main content
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"]
    03/11/2021 3:27 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENTRYPOINT []
    03/11/2021 3:27 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
    03/11/2021 3:27 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CONDA_ENV=rapids HUGECTR_VER=v3.0 NVTAB_VER=v0.4.0 RAPIDS_VER=0.18 RELEASE=true SM=60;61;70;75;80 /bin/bash -c echo $(du -h --max-depth=1 /)
    03/11/2021 3:27 AM UTC
    sha256:76fbc1edf93d228958b3721b14c124a5d7c8327223dda3c75c0fb332a6a2d77cRUN
    CONDA_ENV=rapids HUGECTR_VER=v3.0 NVTAB_VER=v0.4.0 RAPIDS_VER=0.18 RELEASE=true SM=60;61;70;75;80 /bin/bash -c source activate ${CONDA_ENV}; conda clean --all -y
    03/11/2021 3:27 AM UTC
    sha256:0b692b0af7842256201172e2154d069b46a96b2c992f10a811e7344983424ab5RUN
    CONDA_ENV=rapids HUGECTR_VER=v3.0 NVTAB_VER=v0.4.0 RAPIDS_VER=0.18 RELEASE=true SM=60;61;70;75;80 /bin/bash -c source activate ${CONDA_ENV}; apt update; apt install -y graphviz ;
    03/11/2021 3:27 AM UTC
    sha256:da888a25b0a73ff688320d7410d8cd3bd637763de50966ec95d310b0bb5b0923RUN
    CONDA_ENV=rapids HUGECTR_VER=v3.0 NVTAB_VER=v0.4.0 RAPIDS_VER=0.18 RELEASE=true SM=60;61;70;75;80 /bin/bash -c source activate ${CONDA_ENV}; conda env config vars set PYTHONPATH=$PYTHONPATH:/opt/conda/envs/rapids/lib/python3.8/site-packages:/hugectr/tools/embedding_plugin/python:/opt/conda/lib/python3.8/site-packages
    03/11/2021 3:26 AM UTC
    sha256:d828736ae961d7910af5a851fcb01cfeb74828e99cd96bcb63db05e97bef784dRUN
    CONDA_ENV=rapids HUGECTR_VER=v3.0 NVTAB_VER=v0.4.0 RAPIDS_VER=0.18 RELEASE=true SM=60;61;70;75;80 /bin/bash -c source activate ${CONDA_ENV}; conda install -c rapidsai asvdb
    03/11/2021 3:26 AM UTC
    sha256:f95a1a34ec8fd5131996b2b2e9fc3070e9929889420625a509c9133bd5bc3a7dRUN
    CONDA_ENV=rapids HUGECTR_VER=v3.0 NVTAB_VER=v0.4.0 RAPIDS_VER=0.18 RELEASE=true SM=60;61;70;75;80 /bin/bash -c source activate ${CONDA_ENV}; pip install nvidia-pyindex; pip install tritonclient[all] grpcio-channelz
    03/11/2021 3:26 AM UTC
    sha256:1a0a7f6010feaf656f13fd911f5ff778c52282894e295f419c13645231436a1fRUN
    CONDA_ENV=rapids HUGECTR_VER=v3.0 NVTAB_VER=v0.4.0 RAPIDS_VER=0.18 RELEASE=true SM=60;61;70;75;80 /bin/bash -c source activate ${CONDA_ENV}; pip install pynvml pytest graphviz sklearn scipy matplotlib
    03/11/2021 3:26 AM UTC
    ...