Entrypoint /opt/nvidia/nvidia_entrypoint.sh

RUN |2 ENABLE_MITMPROXY=0 VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION /bin/sh -c set -exo pipefail export DEVEL=1 BASE=0 if [ ! -f /etc/ld.so.conf.d/nvidia-tegra.conf ]; then COMPONENT_LIST="nvrtc-dev" /nvidia/build-scripts/installCUDA.sh else # Don't use COMPONENT_LIST in iGPU environment
    /nvidia/build-scripts/installCUDA.sh
fi

VERSION_LIST="${VERSION_LIST}" /nvidia/build-scripts/installLIBS.sh
if [ ! -f /etc/ld.so.conf.d/nvidia-tegra.conf ]; then 
  /nvidia/build-scripts/installNCCL.sh
fi
NOBUILDER=1 /nvidia/build-scripts/installTRT.sh
/nvidia/build-scripts/installCUDNN.sh
/nvidia/build-scripts/installCUSPARSELT.sh

# Clean up libcudnn_engines_precompiled
rm -rf /usr/lib/*-linux-gnu/libcudnn_engines_precompiled.so*
 # buildkit

VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION

ENABLE_MITMPROXY=0 /tmp/manage_cert.sh install

ENABLE_MITMPROXY=0

com.nvidia.nccl.version=2.29.stable.20260109 com.nvidia.cublas.version=13.2.1.1 com.nvidia.cufft.version=12.1.0.78 com.nvidia.curand.version=10.4.1.81 com.nvidia.cusparse.version=12.7.3.1 com.nvidia.cusparselt.version=0.8.1.1 com.nvidia.cusolver.version=12.0.9.81 com.nvidia.npp.version=13.0.3.3 com.nvidia.nvjpeg.version=13.0.3.75 com.nvidia.cudnn.version=9.17.1.4

ENABLE_MITMPROXY=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 |3 ENABLE_MITMPROXY=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 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 if [ ! -f /etc/ld.so.conf.d/nvidia-tegra.conf ]; then /nvidia/build-scripts/installNCCL.sh /nvidia/build-scripts/installNVSHMEM.sh # Manually create symlinks nvshmem so we don't need to use dpkg-divert
  # nvshmem default installation is under /usr/lib/x86_64-linux-gnu/nvshmem/${CUDA_VERSION_MAJOR}
  # We need to libs to be available under /usr/local/cuda/lib64
  export CUDA_VERSION_MAJOR=$(echo "${CUDA_VERSION}" | cut -d. -f1)
  find /usr/lib/*-linux-gnu/nvshmem/${CUDA_VERSION_MAJOR}/ -maxdepth 1 -type f -exec ln -sf {} /usr/local/cuda/lib64/ \;
  # For symlinks under /usr/lib/*-linux-gnu/nvshmem/${CUDA_VERSION_MAJOR}/, we want to create it under /usr/local/cuda/lib64/ and link it to the same target
  find /usr/lib/*-linux-gnu/nvshmem/${CUDA_VERSION_MAJOR}/ -maxdepth 1 -type l -exec ln -sf {} /usr/local/cuda/lib64/ \;
  # Ensure libnvshmem_host.so* files exist under /usr/local/cuda/lib64
  ls /usr/local/cuda/lib64/libnvshmem_host.so* || { echo "[ERROR] libnvshmem_host.so* not found under /usr/local/cuda/lib64"; exit 1; }
fi
 # buildkit

VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION