download nvidia drivers and CUDA libs

nvidia.sh

# This script is built on kops bootstrap
# https://github.com/dcwangmit01/kops/blob/ef958a7f870eb3dc20981617859b8ad69057bb2a/hooks/nvidia-bootstrap/image/run.sh

mkdir nvidia
cd nvidia

DRIVER_VERSION="440.82"
ARCH="Linux-x86_64"
DRIVER_FILE="NVIDIA-${ARCH}-${DRIVER_VERSION}.run"
DRIVER_URL="http://us.download.nvidia.com/XFree86/${ARCH}/${DRIVER_VERSION}/${DRIVER_FILE}"
CUDA_MAJOR_VERSION="10.2"
CUDA_MINOR_VERSION="89_440.33.01"

# Get the NVIDIA driver
curl -LO ${DRIVER_URL}
chmod a+x ${DRIVER_FILE}

# Download CUDA toolkit and patches
cuda_files=( \
  "http://developer.download.nvidia.com/compute/cuda/${CUDA_MAJOR_VERSION}/Prod/local_installers/cuda_${CUDA_MAJOR_VERSION}.${CUDA_MINOR_VERSION}_linux.run" \
  )

for download in ${cuda_files[@]}; do
  curl -LO $download
done

chmod a+x cuda_*.run