Update generate_binary_build_matrix.py Add CUDA 12.6 entry in CUDA_CU…

…DDN_VERSIONS (#5955) Required for Docker build validation
pytorch · Nov 20, 2024 · 674143c · 674143c
1 parent 5ff1c7e
commit 674143c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -39,6 +39,7 @@
     "11.8": {"cuda": "11.8.0", "cudnn": "9"},
     "12.1": {"cuda": "12.1.1", "cudnn": "9"},
     "12.4": {"cuda": "12.4.1", "cudnn": "9"},
+    "12.6": {"cuda": "12.6.2", "cudnn": "9"},
 }
 
 PACKAGE_TYPES = ["wheel", "conda", "libtorch"]