install_cuda_docker_image.sh

#!/bin/bash

set -e

export DEBIAN_FRONTEND=noninteractive

DISTRO=debian10
CUDA_KEYRING_DEB=cuda-keyring_1.0-1_all.deb

wget -O /tmp/${CUDA_KEYRING_DEB} https://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}/x86_64/${CUDA_KEYRING_DEB}
dpkg -i /tmp/${CUDA_KEYRING_DEB}

sed -i 's/bullseye main/bullseye main contrib non-free/' /etc/apt/sources.list

apt-get update -yq

# 11.4 devel (https://hub.docker.com/layers/nvidia/cuda/11.4.0-cudnn8-devel-ubuntu20.04/images/sha256-5f24d101e393aa2c6b91073feef55569a06d1b18c7c6e789ea388f402d1fc2e9?context=explore)
#NV_CUDA_LIB_VERSION=11.4.0-1
#NV_LIBNPP_PACKAGE=libnpp-11-4=11.4.0.33-1
#NV_NVTX_VERSION=11.4.43-1
#NV_LIBCUSPARSE_VERSION=11.6.0.43-1
#NV_LIBCUBLAS_PACKAGE=libcublas-11-4=11.5.2.43-1
#NV_LIBNCCL_PACKAGE=libnccl2=2.11.4-1+cuda11.4
#NV_CUDNN_PACKAGE=libcudnn8=8.2.4.15-1+cuda11.4
#NV_CUDNN_PACKAGE_DEV=libcudnn8-dev=8.2.4.15-1+cuda11.4
#
#apt-get update && apt-get install -y --no-install-recommends cuda-libraries-11-4=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} \
#  cuda-nvtx-11-4=${NV_NVTX_VERSION} libcusparse-11-4=${NV_LIBCUSPARSE_VERSION} ${NV_LIBCUBLAS_PACKAGE} ${NV_LIBNCCL_PACKAGE}
#apt-get update && apt-get install -y --no-install-recommends ${NV_CUDNN_PACKAGE} ${NV_CUDNN_PACKAGE_DEV}

# 11.4 runtime (https://hub.docker.com/layers/nvidia/cuda/11.4.0-cudnn8-runtime-ubuntu20.04/images/sha256-032726ef96a82a49f4a2f5a48e2035c78aeff7b34213003516c9077cc2c61f47?context=explore)
#NV_CUDA_LIB_VERSION=11.4.0-1
#NV_LIBNPP_PACKAGE=libnpp-11-4=11.4.0.33-1
#NV_NVTX_VERSION=11.4.43-1
#NV_LIBCUSPARSE_VERSION=11.6.0.43-1
#NV_LIBCUBLAS_PACKAGE=libcublas-11-4=11.5.2.43-1
##NV_LIBNCCL_PACKAGE=libnccl2=2.11.4-1+cuda11.4
##NV_CUDNN_PACKAGE=libcudnn8=8.2.4.15-1+cuda11.4
#
#apt-get install -y --no-install-recommends cuda-libraries-11-4=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} \
#  cuda-nvtx-11-4=${NV_NVTX_VERSION} libcusparse-11-4=${NV_LIBCUSPARSE_VERSION} ${NV_LIBCUBLAS_PACKAGE} #${NV_LIBNCCL_PACKAGE}
##apt-get install -y --no-install-recommends ${NV_CUDNN_PACKAGE}

echo "/usr/local/nvidia/lib" >>/etc/ld.so.conf.d/nvidia.conf
echo "/usr/local/nvidia/lib64" >>/etc/ld.so.conf.d/nvidia.conf

apt-get install -y --no-install-recommends nvidia-smi

apt-get clean
apt-get autoclean
rm -rf /var/lib/apt/lists/*