Update nvidia container support package (#688)

* Removed deprecated package and installed nvidia-container-toolkit

* Removing apt-get in favor of apt

* Updated golden tests

Co-authored-by: Helio Machado <0x2b3bfa0+git@googlemail.com>

Author: aliabbasjaffri

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-docker2
+    sudo apt update && sudo apt install -y nvidia-container-toolkit
     sudo systemctl restart docker
   fi
 

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-docker2
+    sudo apt update && sudo apt install -y nvidia-container-toolkit
     sudo systemctl restart docker
   fi
 

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-docker2
+    sudo apt update && sudo apt install -y nvidia-container-toolkit
     sudo systemctl restart docker
   fi
 

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-docker2
+    sudo apt update && sudo apt install -y nvidia-container-toolkit
     sudo systemctl restart docker
   fi
 

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-docker2
+    sudo apt update && sudo apt install -y nvidia-container-toolkit
     sudo systemctl restart docker
   fi