Revert "Update nvidia container support package (#688)" (#690)

This reverts commit c574e4a.
iterative · Oct 8, 2022 · 5f6b95d · 5f6b95d
1 parent c574e4a
commit 5f6b95d
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/environment/setup.sh b/environment/setup.sh
@@ -37,7 +37,7 @@ if [ ! -f "$FILE" ]; then
 
     curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
     curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
-    sudo apt update && sudo apt install -y nvidia-container-toolkit
+    sudo apt update && sudo apt install -y nvidia-docker2
     sudo systemctl restart docker
   fi
 

diff --git a/iterative/testdata/script_template_cloud_aws.golden b/iterative/testdata/script_template_cloud_aws.golden
@@ -38,7 +38,7 @@ if [ ! -f "$FILE" ]; then
 
     curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
     curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
-    sudo apt update && sudo apt install -y nvidia-container-toolkit
+    sudo apt update && sudo apt install -y nvidia-docker2
     sudo systemctl restart docker
   fi
 

diff --git a/iterative/testdata/script_template_cloud_azure.golden b/iterative/testdata/script_template_cloud_azure.golden
@@ -38,7 +38,7 @@ if [ ! -f "$FILE" ]; then
 
     curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
     curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
-    sudo apt update && sudo apt install -y nvidia-container-toolkit
+    sudo apt update && sudo apt install -y nvidia-docker2
     sudo systemctl restart docker
   fi
 

diff --git a/iterative/testdata/script_template_cloud_gcp.golden b/iterative/testdata/script_template_cloud_gcp.golden
@@ -38,7 +38,7 @@ if [ ! -f "$FILE" ]; then
 
     curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
     curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
-    sudo apt update && sudo apt install -y nvidia-container-toolkit
+    sudo apt update && sudo apt install -y nvidia-docker2
     sudo systemctl restart docker
   fi
 

diff --git a/iterative/testdata/script_template_cloud_invalid.golden b/iterative/testdata/script_template_cloud_invalid.golden
@@ -38,7 +38,7 @@ if [ ! -f "$FILE" ]; then
 
     curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
     curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
-    sudo apt update && sudo apt install -y nvidia-container-toolkit
+    sudo apt update && sudo apt install -y nvidia-docker2
     sudo systemctl restart docker
   fi