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"]
    05/28/2021 9:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENTRYPOINT []
    05/28/2021 9:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
    05/28/2021 9:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.3 RELEASE=true RMM_VER=v0.19.0 SM=60;61;70;75;80 /bin/bash -c echo $(du -h --max-depth=1 /)
    05/28/2021 9:35 AM UTC
    sha256:71ac2a1f8f656f3589fdc1031e7f9444c7357b5191e3a74b38343e53b8333d29RUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.3 RELEASE=true RMM_VER=v0.19.0 SM=60;61;70;75;80 /bin/bash -c pip uninstall numpy -y; pip install numpy
    05/28/2021 9:34 AM UTC
    sha256:4d1baa54bb1eca2e0e490c05f0e3e3c5d0651b171ca6ea777d8342a7afdc7df3RUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.3 RELEASE=true RMM_VER=v0.19.0 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
    05/28/2021 9:34 AM 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:
    05/28/2021 9:34 AM UTC
    sha256:bca0a06ce5e219205048315a4da754feb1d6fd2db9ea572979885392b3349aa0RUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.3 RELEASE=true RMM_VER=v0.19.0 SM=60;61;70;75;80 /bin/bash -c apt-get update; apt-get install -y graphviz
    05/28/2021 9:34 AM UTC
    sha256:2f02be6f5b68b65be2b5bf6a63119738caa14a65f380fb97b39e1f396cfae69eRUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.3 RELEASE=true RMM_VER=v0.19.0 SM=60;61;70;75;80 /bin/bash -c pip install dask==2021.04.0 nvtx pandas==1.1.5 mpi4py==3.0.3 cupy-cuda112 cachetools typing_extensions fastavro
    05/28/2021 9:34 AM UTC
    sha256:29da7398c2120bb899bca883bffb84cc39fd9f68ec4f37b9d79117ff0c4a9f2cRUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.3 RELEASE=true RMM_VER=v0.19.0 SM=60;61;70;75;80 /bin/bash -c pip install nvidia-pyindex; pip install tritonclient[all] grpcio-channelz
    05/28/2021 9:32 AM 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.