Entrypoint /opt/nvidia/nvidia_entrypoint.sh

ENABLE_MITMPROXY=0 ENABLE_FIPS=0 CUDA_COMPONENT_LIST=cccl crt nvrtc driver-dev culibos-dev cudart cudart-dev nvcc tileiras VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION /tmp/manage_cert.sh uninstall

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

RUN |4 ENABLE_MITMPROXY=0 ENABLE_FIPS=0 CUDA_COMPONENT_LIST=cccl crt nvrtc driver-dev culibos-dev cudart cudart-dev nvcc tileiras VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION /bin/sh -c set -exo pipefail export ARTIFACTORY_USER=$(cat /run/secrets/ARTIFACTORY_USER) export ARTIFACTORY_TOKEN=$(cat /run/secrets/ARTIFACTORY_TOKEN) VERSION_LIST="${VERSION_LIST}" /nvidia/build-scripts/installLIBS.sh # Hack to grab stubs from the libs stage and put them under /usr/local/cuda/lib64 without actually installing the libs HACK_LIST="cusparse cusolver cufft nvJitLink" for lib in ${HACK_LIST}; do _LIB_VERSION_ENV_VAR="${lib^^}_VERSION" _LIB_MAJOR_VERSION=$(echo "${!_LIB_VERSION_ENV_VAR}" | cut -d. -f1) if [[ -z "${_LIB_MAJOR_VERSION}" ]]; then echo "[ERROR] ${_LIB_VERSION_ENV_VAR} is not set" exit 1 fi for _stub in $(ls /tmp/stubs/*${lib}*); do cp ${_stub} /usr/local/cuda/lib64/$(basename ${_stub}).${_LIB_MAJOR_VERSION} # Create a symlink to the file with .0 extension ln -sf $(basename ${_stub}).${_LIB_MAJOR_VERSION} /usr/local/cuda/lib64/$(basename ${_stub}).0 done done ( set +x; echo "[INFO] /usr/local/cuda/lib64 after hack:"; ls -larth /usr/local/cuda/lib64/ ) /nvidia/build-scripts/installCUDNN.sh /nvidia/build-scripts/installCUSPARSELT.sh NOBUILDER=1 /nvidia/build-scripts/installTRT.sh /nvidia/build-scripts/installNCCL.sh DPKG_DIVERT=1 /nvidia/build-scripts/installNVSHMEM.sh ls /usr/local/cuda/lib64/libnvshmem_host.so* ||
  { echo "[ERROR] libnvshmem_host.so* not found under /usr/local/cuda/lib64"; exit 1; } # buildkit

VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION

RUN |3 ENABLE_MITMPROXY=0 ENABLE_FIPS=0 CUDA_COMPONENT_LIST=cccl crt nvrtc driver-dev culibos-dev cudart cudart-dev nvcc tileiras /bin/sh -c set -exo pipefail export ARTIFACTORY_USER=$(cat /run/secrets/ARTIFACTORY_USER) export ARTIFACTORY_TOKEN=$(cat /run/secrets/ARTIFACTORY_TOKEN) COMPONENT_LIST="${CUDA_COMPONENT_LIST}" /nvidia/build-scripts/installCUDA.sh # buildkit

CUDA_COMPONENT_LIST=cccl crt nvrtc driver-dev culibos-dev cudart cudart-dev nvcc tileiras

CUDA_COMPONENT_LIST=cccl crt nvrtc driver-dev culibos-dev cudart cudart-dev nvcc tileiras

ENABLE_MITMPROXY=0 ENABLE_FIPS=0 export DEBIAN_FRONTEND=noninteractive &&
  apt-get update &&
  apt-get install -y --no-install-recommends apt-utils build-essential libncurses6 libncursesw6 unzip jq gnupg libtcmalloc-minimal4 git libglib2.0-0 libhwloc15 libnuma-dev libnuma1 libpmi2-0-dev numactl openssh-client vim-tiny wget &&
  rm -rf /var/lib/apt/lists/* &&
  if [ "${ENABLE_FIPS}" = "1" ]; 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 )

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