Skip to main content
NVIDIA
Merlin Training
Container
NVIDIA
Merlin Training

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

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) CMD ["/bin/bash"]
    01/14/2022 9:17 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENTRYPOINT []
    01/14/2022 9:17 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
    01/14/2022 9:17 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    HUGECTR_VER=vnightly HWLOC_VER=2.4.1 NVTAB_VER=vnightly RELEASE=false /bin/bash -c echo $(du -h --max-depth=1 /)
    01/14/2022 9:17 AM UTC
    sha256:2788fd6a18393b9215c3a880c4c2ffd96f2722b8ce86661532213745346da344RUN
    HUGECTR_VER=vnightly HWLOC_VER=2.4.1 NVTAB_VER=vnightly RELEASE=false /bin/bash -c pip install numba numpy --upgrade
    01/14/2022 9:17 AM UTC
    sha256:b3126ef3a509c3e0164867531d45b8c9c01367ce222017834184868c720f3308RUN
    HUGECTR_VER=vnightly HWLOC_VER=2.4.1 NVTAB_VER=vnightly RELEASE=false /bin/bash -c pip install dask==2021.09.1 distributed==2021.09.1 dask[dataframe]==2021.09.1 dask-cuda
    01/14/2022 9:17 AM UTC
    sha256:e7e95d12cf6fd695c9c164689ab6221640c342c564194d2e9586eaa0a6671185RUN
    HUGECTR_VER=vnightly HWLOC_VER=2.4.1 NVTAB_VER=vnightly RELEASE=false /bin/bash -c rm -rf /repos
    01/14/2022 9:17 AM UTC
    sha256:d94195d5caf8112cd776c24ea704c63f6d4ad4eaa967da1935a322d3f7a8eb43RUN
    HUGECTR_VER=vnightly HWLOC_VER=2.4.1 NVTAB_VER=vnightly RELEASE=false /bin/bash -c rm /usr/local/cuda/lib64/stubs/libcuda.so.1
    01/14/2022 9:17 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENV PYTHONPATH=/hugectr/onnx_converter:/usr/local/hugectr/lib:
    01/14/2022 9:17 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENV PYTHONPATH=/usr/local/hugectr/lib:
    01/14/2022 9:17 AM UTC
    ...