remove nvidia-docker2 (#6943)

Follow up to remove empty container nvidia-docker2 and change one more
version for ubuntu containers

tested with:
pytorch/pytorch#158684

Author: wdvr

.github/actions/setup-nvidia/action.yml

@@ -7,7 +7,7 @@ inputs:
     description: which driver version to install
     required: false
     type: string
-    default: "570.133.07"   # https://www.nvidia.com/en-us/drivers/details/242273
+    default: "570.133.07" # https://www.nvidia.com/en-us/drivers/details/242273
 
 runs:
   using: composite
@@ -41,7 +41,6 @@ runs:
 
                   sudo yum-config-manager --add-repo "${YUM_REPO_URL}"
                   sudo yum install -y \
-                    nvidia-docker2 \
                     nvidia-container-toolkit-1.17.8 \
                     libnvidia-container-tools-1.17.8 \
                     libnvidia-container1-1.17.8 \
@@ -56,7 +55,7 @@ runs:
                   # Install nvidia-driver package if not installed
                   status="$(dpkg-query -W --showformat='${db:Status-Status}' nvidia-docker2 2>&1)"
                   if [ ! $? = 0 ] || [ ! "$status" = installed ]; then
-                    sudo apt-get install -y nvidia-docker2 nvidia-container-toolkit-1.16.2
+                    sudo apt-get install -y nvidia-container-toolkit-1.17.8
                     sudo systemctl restart docker
                   fi
               )