Skip to content

Commit

Permalink
Add nvidia-docker tools installation to worker provisioning (#4440)
Browse files Browse the repository at this point in the history
follow up on prev PR
  • Loading branch information
atalman authored Jul 28, 2023
1 parent fb7ad95 commit 7e640b2
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ echo Checking if nvidia install required ${nvidia_driver_install}
%{ if nvidia_driver_install ~}
set +e
os_id=$(. /etc/os-release;echo $ID$VERSION_ID)
if ! [[ "$os_id" =~ ^amzn.* ]]; then
if [[ "$os_id" =~ ^amzn.* ]]; then
echo Installing Development Tools
sudo yum groupinstall -y "Development Tools"
sudo yum install -y "kernel-devel-uname-r == $(uname -r)"
Expand All @@ -34,7 +34,7 @@ fi
sudo curl -fsL -o /tmp/nvidia_driver "https://s3.amazonaws.com/ossci-linux/nvidia_driver/NVIDIA-Linux-x86_64-535.54.03.run"
sudo /bin/bash /tmp/nvidia_driver -s --no-drm
sudo rm -fv /tmp/nvidia_driver
if ! [[ "$os_id" =~ ^amzn.* ]]; then
if [[ "$os_id" =~ ^amzn.* ]]; then
echo Installing nvidia-docker tools
sudo yum install -y yum-utils
sudo yum-config-manager --add-repo https://nvidia.github.io/nvidia-docker/$os_id/nvidia-docker.repo
Expand Down

0 comments on commit 7e640b2

Please sign in to comment.