Skip to content
This repository was archived by the owner on Dec 4, 2023. It is now read-only.

issue-#202-resolved with nvidia-container install #9

Open
ImayaBharathi wants to merge 2 commits into
base: main
Choose a base branch
from
Open

issue-#202-resolved with nvidia-container install #9

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ad1e006
issue-#202-resolved with nvidia-container install
ImayaBharathi Jun 29, 2023
2111a1e
Update install.sh - increased SCRIPT_VERSION to 0.0.2
ImayaBharathi Jul 2, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 18 additions & 1 deletion install.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Copy link
Contributor

@tiero tiero Jul 1, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please increase the script version to 0.0.2 at the top of the file

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
Expand Down Expand Up @@ -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
Expand Down