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"]
    05/04/2021 1:30 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENTRYPOINT []
    05/04/2021 1:30 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
    05/04/2021 1:30 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.1 RELEASE=true RMM_VER=v0.19.0 SM=60;61;70;75;80 /bin/bash -c echo $(du -h --max-depth=1 /)
    05/04/2021 1:30 AM UTC
    sha256:b1394251b1be5254a7aa4607739c8b6e28f785d3a3b7956b11a047e90f8030e8RUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.1 RELEASE=true RMM_VER=v0.19.0 SM=60;61;70;75;80 /bin/bash -c pip uninstall numpy -y; pip install numpy
    05/04/2021 1:30 AM UTC
    sha256:889a91fd5fe5a5fdbe06752b31337116f7ca365ee2e565b2d7b1cafad42beb5aRUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.1 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/04/2021 1:30 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/04/2021 1:29 AM UTC
    sha256:d0e6214ec08dda00948964604999abf381b80ef994ce8c183508413b269721fcRUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.1 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/04/2021 1:29 AM UTC
    sha256:b8a2ff7f7f2a2c2448eac5b11c471410a0da3f5065a572abcd97c4a8339e959dRUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.1 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/04/2021 1:29 AM UTC
    sha256:7f167147f61d408d220f4d4027b912c06f162c8b4bc24f697ac0fd5dda274b68RUN
    CUDF_VER=v0.19.1 HUGECTR_VER=v3.0.1 NVTAB_VER=v0.5.1 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/04/2021 1:28 AM UTC
    ...