install into hiveos to allow GPU's to be passed to docker container

hive_os_install.sh

#!/bin/bash

set -x

apt update -y && apt install gpg sed curl sudo  -y

nvidia-driver-update --version 535.113.01

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
      && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
      && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
            sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
            sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list


apt-get update \
    && sudo apt-get install -y\
    nvidia-container-toolkit-base \
    docker.io  \
    nvidia-docker2


nvidia-ctk runtime configure --runtime=docker
nvidia-ctk cdi generate --output=/etc/cdi/nvidia.yaml
systemctl restart docker


#
# Verify everything works with the command
#
docker run --rm  --gpus all nvidia/cuda:12.0.1-base-ubuntu20.04 nvidia-smi