Entrypoint /opt/nvidia/nvidia_entrypoint.sh

LIBRARY_PATH=/usr/local/cuda/lib64/stubs:/usr/local/cuda/lib64/stubs:

RUN |3 TARGETARCH=amd64 ENABLE_FIPS=0 ENABLE_MITMPROXY=0 /bin/sh -c set -exo pipefail export ARTIFACTORY_USER=$(cat /run/secrets/ARTIFACTORY_USER) export ARTIFACTORY_TOKEN=$(cat /run/secrets/ARTIFACTORY_TOKEN) export DEVEL=1 BASE=0 /nvidia/build-scripts/installNCU.sh /nvidia/build-scripts/installCUDA.sh /nvidia/build-scripts/installLIBS.sh /nvidia/build-scripts/installNCCL.sh /nvidia/build-scripts/installNVSHMEM.sh export CUDA_VERSION_MAJOR=$(echo "${CUDA_VERSION}" | cut -d. -f1) # Link nvshmem libs to /usr/local/cuda/lib64 find /usr/lib/*-linux-gnu/nvshmem/${CUDA_VERSION_MAJOR}/ -maxdepth 1 -type f -exec ln -sf {} /usr/local/cuda/lib64/ \; find /usr/lib/*-linux-gnu/nvshmem/${CUDA_VERSION_MAJOR}/ -maxdepth 1 -type l -exec ln -sf {} /usr/local/cuda/lib64/ \; /nvidia/build-scripts/installCUDNN.sh /nvidia/build-scripts/installTRT.sh ARTIFACTORY_CLOUD=1 /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 # buildkit

cuda-*.patch /tmp

TARGETARCH=amd64 ENABLE_FIPS=0 ENABLE_MITMPROXY=0 export DEBIAN_FRONTEND=noninteractive &&
  apt-get update &&
  apt-get install -y --no-install-recommends build-essential git libglib2.0-0 less libhwloc15 libnuma-dev libnuma1 libpmi2-0-dev nano numactl openssh-client vim wget &&
  rm -rf /var/lib/apt/lists/* &&
  if [ "${ENABLE_FIPS}" = "1" -a "${TARGETARCH}" = "amd64" ]; then apt-get remove -y openssh-client &&
  cp -f -p /tmp/ubuntu.sources.fips /etc/apt/sources.list.d/ubuntu.sources &&
  apt-get update &&
  apt-get install -y --no-install-recommends openssh-client &&
  rm -rf /var/lib/apt/lists/*; fi &&
  ( [ -x /bin/ssh-agent ] &&
  chgrp root /bin/ssh-agent ||
  true ) &&
  ( [ -x /usr/bin/ssh-agent ] &&
  chgrp root /usr/bin/ssh-agent ||
  true )

TARGETARCH=amd64 ENABLE_FIPS=0 ENABLE_MITMPROXY=0 /tmp/manage_cert.sh install

ENABLE_MITMPROXY=0

ENABLE_FIPS=0

TARGETARCH=amd64

ENABLE_MITMPROXY=0 /tmp/manage_cert.sh uninstall