ci(torch): Build against CUDA 12.6; remove 12.3, 12.1, and lower builds

Additionally, this change switches to base images with updated
NCCL (up to v2.23.4-1) and HPC-X (v2.20) versions for torch:nccl.

.github/configurations/torch-base.yml

@@ -1,15 +1,5 @@
-cuda: [ 12.4.1, 12.3.2, 12.2.2, 12.0.1, 11.8.0 ]
+cuda: [ 12.6.1, 12.4.1, 12.2.2 ]
 os: [ ubuntu22.04, ubuntu20.04 ]
-exclude:
-  # Not a supported combination
-  - cuda: 11.8.0
-    os: ubuntu22.04
-  - cuda: 11.8.0
-    os: ubuntu20.04
-  - cuda: 12.0.1
-    os: ubuntu20.04
-  - cuda: 12.0.1
-    os: ubuntu22.04
 include:
   - torch: 2.4.1
     vision: 0.19.1

.github/configurations/torch-nccl.yml

@@ -1,51 +1,36 @@
 image:
   # Ubuntu 22.04
-  - cuda: 12.4.1
+  - cuda: 12.6.1
     cudnn: cudnn
     os: ubuntu22.04
-    nccl: 2.21.5-1
-    nccl-tests-hash: 85f9143
-  - cuda: 12.3.2
-    cudnn: cudnn9
+    nccl: 2.23.4-1
+    nccl-tests-hash: 2ff05b2
+  - cuda: 12.4.1
+    cudnn: cudnn
     os: ubuntu22.04
-    nccl: 2.20.3-1
-    nccl-tests-hash: 85f9143
+    nccl: 2.23.4-1
+    nccl-tests-hash: 2ff05b2
   - cuda: 12.2.2
     cudnn: cudnn8
     os: ubuntu22.04
-    nccl: 2.19.3-1
-    nccl-tests-hash: 85f9143
-#  - cuda: 12.0.1
-#    cudnn: cudnn8
-#    os: ubuntu22.04
-#    nccl: 2.18.5-1
-#    nccl-tests-hash: 85f9143
+    nccl: 2.23.4-1
+    nccl-tests-hash: 2ff05b2
   # Ubuntu 20.04
-  - cuda: 12.4.1
+  - cuda: 12.6.1
     cudnn: cudnn
     os: ubuntu20.04
-    nccl: 2.21.5-1
-    nccl-tests-hash: 85f9143
-  - cuda: 12.3.2
-    cudnn: cudnn9
+    nccl: 2.23.4-1
+    nccl-tests-hash: 2ff05b2
+  - cuda: 12.4.1
+    cudnn: cudnn
     os: ubuntu20.04
-    nccl: 2.20.3-1
-    nccl-tests-hash: 85f9143
+    nccl: 2.23.4-1
+    nccl-tests-hash: 2ff05b2
   - cuda: 12.2.2
     cudnn: cudnn8
     os: ubuntu20.04
     nccl: 2.21.5-1
-    nccl-tests-hash: 85f9143
-#  - cuda: 12.0.1
-#    cudnn: cudnn8
-#    os: ubuntu20.04
-#    nccl: 2.19.3-1
-#    nccl-tests-hash: 85f9143
-#  - cuda: 11.8.0
-#    cudnn: cudnn8
-#    os: ubuntu20.04
-#    nccl: 2.16.5-1
-#    nccl-tests-hash: 868dc3d
+    nccl-tests-hash: 2ff05b2
 include:
   - torch: 2.4.1
     vision: 0.19.1

torch/Dockerfile

@@ -1,6 +1,6 @@
 # syntax=docker/dockerfile:1.4
-ARG BUILDER_BASE_IMAGE="nvidia/cuda:12.0.1-devel-ubuntu22.04"
-ARG FINAL_BASE_IMAGE="nvidia/cuda:12.0.1-base-ubuntu22.04"
+ARG BUILDER_BASE_IMAGE="nvidia/cuda:12.4.1-devel-ubuntu22.04"
+ARG FINAL_BASE_IMAGE="nvidia/cuda:12.4.1-base-ubuntu22.04"
 
 ARG BUILD_TORCH_VERSION="2.4.1"
 ARG BUILD_TORCH_VISION_VERSION="0.19.1"