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"]
    09/25/2021 9:32 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENTRYPOINT []
    09/25/2021 9:32 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
    09/25/2021 9:32 PM UTC
    sha256:6536d4263d7590248ac2659e545350570aa55871d2d7b36bd30711089b77504eRUN
    HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c echo $(du -h --max-depth=1 /)
    09/25/2021 9:32 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) SHELL [/bin/bash -c]
    09/25/2021 9:32 PM UTC
    sha256:140de424640898b395be478a1c7962f72c6d0962f03574bb94ccbd007d04e92fRUN
    HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c pip install numba numpy --upgrade
    09/25/2021 9:32 PM UTC
    sha256:db9061fee435594828666bdc9a728bccfac54f281878d3cbdc663ef39f6f342dRUN
    HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c pip install pybind11
    09/25/2021 9:32 PM UTC
    sha256:14630817fb063a981a8ea66e47f81c568df0d4b9c2f91ca3ed2a4dece2f75506RUN
    HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c git clone https://github.com/NVIDIA/HugeCTR.git build-env &&
      pushd build-env &&
      if [ "$RELEASE" == "true" ] &&
      [ ${HUGECTR_VER} != "vnightly" ] ; then git fetch --all --tags &&
      git checkout tags/${HUGECTR_VER}; else echo ${HUGECTR_VER} &&
      git checkout ${HUGECTR_VER}-integration; fi &&
      cd sparse_operation_kit &&
      bash ./install.sh --SM=$SM --USE_NVTX=$USE_NVTX &&
      popd &&
      rm -rf build-env &&
      rm -rf /var/tmp/HugeCTR
    09/25/2021 9:32 PM UTC
    sha256:d4c089d34c2623537082b578e376ce9dfb7916ce1fc59089fc2ab61d2c6d91d4RUN
    HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c ln -s /usr/lib/x86_64-linux-gnu/libibverbs.so.1.11.32.1 /usr/lib/x86_64-linux-gnu/libibverbs.so
    09/25/2021 9:30 PM UTC
    sha256:fda42af4eb407907ce7f916964df5ac0d3b7a047ad4e9219ad9106d4524aae33RUN
    HUGECTR_VER=v3.2 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c mkdir -p /usr/local/nvidia/lib64 &&
      ln -s /usr/local/cuda/lib64/libcusolver.so /usr/local/nvidia/lib64/libcusolver.so.10
    09/25/2021 9:30 PM UTC
    ...