#!/bin/bash

if [ ${DIB_DEBUG_TRACE:-0} -gt 0 ]; then
    set -x
fi
set -o errexit
set -o nounset
set -o pipefail


echo "#############################################################"
echo "# CUDA Installation                                         #"
echo "#############################################################"

# comparing chroot kernel with host kernel
CHROOT_KERNEL=$(dpkg -l | awk '/^.i +linux-(im)/&& $2~/[0-9]/{print substr($2,13)}')
HOST_KERNEL=$(uname -r)
if [ "$CHROOT_KERNEL" != "$HOST_KERNEL" ]; then
	"ERROR: kernel mismatch!"
	exit 1	
fi

#download and install cuda keyring file

mkdir -p /tmp/cuda 
cd /tmp/cuda

if [ -n "${DIB_CUDA_URL}" ]; then
    wget -nv $DIB_CUDA_URL
else
    echo "ERROR: DIB_CUDA_URL is not set."
fi

# Install cuda
unset ARCH
sudo dpkg -i cuda-keyring*
sudo apt-get -y update
sudo apt-get -y install cuda

# clean
export ARCH=amd64
rm -Rf /tmp/cuda
