From ad1e0064424d9ade66e0c4a7d632ab8e9403cecd Mon Sep 17 00:00:00 2001 From: ImayaBharathi Date: Thu, 29 Jun 2023 21:22:04 +0530 Subject: [PATCH 1/2] issue-#202-resolved with nvidia-container install --- install.sh | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/install.sh b/install.sh index 9160a1e..606cf17 100644 --- a/install.sh +++ b/install.sh @@ -54,6 +54,16 @@ PREM_AUTO_UPDATE=$PREM_AUTO_UPDATE" >$PREM_CONF_FOUND curl --silent https://raw.githubusercontent.com/$USER/$REPO/main/docker-compose.gpu.yml -o ~/prem/docker-compose.gpu.yml curl --silent https://raw.githubusercontent.com/$USER/$REPO/main/Caddyfile -o ~/prem/Caddyfile } +# variable and install function for Nvidia-Container Toolkit +distribution=$(. /etc/os-release;echo $ID$VERSION_ID) +echo $distribution +installNvidiaContainerToolkit(){ + curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo apt-key add - + curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list + DEBIAN_FRONTEND=noninteractive sudo apt-get -qq update -y + DEBIAN_FRONTEND=noninteractive sudo apt-get install -qq -y nvidia-docker2 + sudo systemctl restart docker +} # Making base directory for prem if [ ! -d ~/prem ]; then @@ -203,6 +213,13 @@ export SENTRY_DSN=${SENTRY_DSN} export PREM_REGISTRY_URL=${PREM_REGISTRY_URL} # Check if nvidia-smi is available if command -v nvidia-smi > /dev/null 2>&1; then + if [ $(which nvidia-container-toolkit) ]; then + echo "nvidia-container toolkit is available" + else + echo "nvidia-container toolkit is needed for GPU usage inside a container" + echo "Installing nvidia-container toolkit" + installNvidiaContainerToolkit + fi echo "nvidia-smi is available. Running docker-compose.gpu.yml" docker-compose -f ~/prem/docker-compose.yml -f ~/prem/docker-compose.gpu.yml up -d else From 2111a1e3aa851dc86f28fbd4bf16d83ff94bc2c1 Mon Sep 17 00:00:00 2001 From: Imaya Bharathi <32698335+ImayaBharathi@users.noreply.github.com> Date: Sun, 2 Jul 2023 11:21:07 +0530 Subject: [PATCH 2/2] Update install.sh - increased SCRIPT_VERSION to 0.0.2 --- install.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/install.sh b/install.sh index 606cf17..8ab1110 100644 --- a/install.sh +++ b/install.sh @@ -7,7 +7,7 @@ set -eou pipefail PREM_REGISTRY_URL=https://raw.githubusercontent.com/premAI-io/prem-registry/main/manifests.json SENTRY_DSN=https://75592545ad6b472e9ad7c8ff51740b73@o1068608.ingest.sentry.io/4505244431941632 -SCRIPT_VERSION="v0.0.1" +SCRIPT_VERSION="v0.0.2" USER=premai-io REPO=prem-box