#!/bin/bash
#
# Version is given in the CUDA variable.

set -eux

retry() {
    local i=0
    while [ $i -lt 3 ]; do
        if "$@"; then
            return 0
        fi
        i=$((i + 1))
    done
    return 1
}

echo "Installing CUDA support"

retry wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb
retry sudo apt-get update -qq

cuda_prefix=${CUDA:0:4}
cuda_prefix=${cuda_prefix/./-}
retry sudo apt-get install --allow-unauthenticated -y \
    cuda-command-line-tools-${cuda_prefix} \
    libcurand-dev-${cuda_prefix}

retry sudo apt-get clean

cuda_home=/usr/local/cuda-${CUDA:0:4}
$cuda_home/bin/nvcc --version
echo "${cuda_home}/bin" >>$GITHUB_PATH
