NVIDIA
NVIDIA
CUDA Deep Learning
Container
NVIDIA
NVIDIA
CUDA Deep Learning

CUDA is a parallel computing platform and programming model that enhances computing performance using NVIDIA GPUs. CUDA Deep Learning integrates networking and GPU-accelerated libraries like cuDNN, cuTensor, NCCL, HPC-x, and the CUDA Toolkit.

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LIBRARY_PATH=/usr/local/cuda/lib64/stubs:
    09/02/2024 5:26 PM UTC
    sha256:7008501aa4ca4d9bd71cd951739fe9eb9a8bd1ec0ca4c3b69207b3c66206063fRUN
    GDRCOPY_VERSION=2.3.1-1 HPCX_VERSION=2.20 RDMACORE_VERSION=39.0 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.17.0 OPENMPI_VERSION=4.1.7 TARGETARCH=amd64 export DEVEL=1 BASE=0 &&
      /nvidia/build-scripts/installNCU.sh &&
      /nvidia/build-scripts/installCUDA.sh &&
      /nvidia/build-scripts/installLIBS.sh &&
      if [ ! -f /etc/ld.so.conf.d/nvidia-tegra.conf ]; then /nvidia/build-scripts/installNCCL.sh; fi &&
      /nvidia/build-scripts/installCUDNN.sh &&
      /nvidia/build-scripts/installCUTENSOR.sh &&
      /nvidia/build-scripts/installTRT.sh &&
      /nvidia/build-scripts/installNSYS.sh &&
      /nvidia/build-scripts/installCUSPARSELT.sh &&
      if [ -f "/tmp/cuda-${_CUDA_VERSION_MAJMIN}.patch" ]; then patch -p0 < /tmp/cuda-${_CUDA_VERSION_MAJMIN}.patch; fi &&
      rm -f /tmp/cuda-*.patch
    09/02/2024 5:26 PM UTC
    sha256:690875466ef2188c346f57a91e01ce0ac7b3edb9455c710634f75c7addde5f8fCOPY
    cuda-*.patch /tmp
    09/02/2024 5:21 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    OMPI_MCA_coll_hcoll_enable=0
    09/02/2024 5:21 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    OPAL_PREFIX=/opt/hpcx/ompi PATH=/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin
    09/02/2024 5:21 PM UTC
    sha256:720ce94231f25b2ba873e44840c028a17ac9c5d9fcc1fe9b49b8743ae7476755RUN
    GDRCOPY_VERSION=2.3.1-1 HPCX_VERSION=2.20 RDMACORE_VERSION=39.0 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.17.0 OPENMPI_VERSION=4.1.7 TARGETARCH=amd64 cd /nvidia &&
      ( export DEBIAN_FRONTEND=noninteractive &&
      apt-get update &&
      apt-get install -y --no-install-recommends libibverbs1 libibverbs-dev librdmacm1 librdmacm-dev libibumad3 libibumad-dev ibverbs-utils ibverbs-providers &&
      rm -rf /var/lib/apt/lists/* &&
      rm $(dpkg-query -L libibverbs-dev librdmacm-dev libibumad-dev | grep "\(\.so\|\.a\)$") ) &&
      ( cd opt/gdrcopy/ &&
      dpkg -i libgdrapi_*.deb ) &&
      ( cp -r opt/hpcx /opt/ &&
      cp etc/ld.so.conf.d/hpcx.conf /etc/ld.so.conf.d/ &&
      ln -sf /opt/hpcx/ompi /usr/local/mpi &&
      ln -sf /opt/hpcx/ucx /usr/local/ucx &&
      sed -i 's/^\(hwloc_base_binding_policy\) = core$/\1 = none/' /opt/hpcx/ompi/etc/openmpi-mca-params.conf &&
      sed -i 's/^\(btl = self\)$/#\1/' /opt/hpcx/ompi/etc/openmpi-mca-params.conf ) &&
      ldconfig
    09/02/2024 5:21 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TARGETARCH=amd64
    09/02/2024 5:21 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    GDRCOPY_VERSION=2.3.1-1 HPCX_VERSION=2.20 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.17.0 OPENMPI_VERSION=4.1.7 RDMACORE_VERSION=39.0
    09/02/2024 5:21 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    OPENMPI_VERSION=4.1.7
    09/02/2024 5:21 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    OPENUCX_VERSION=1.17.0
    09/02/2024 5:21 PM UTC