build and verify with CUDA 12.6

cupy · Aug 20, 2024 · c76f364 · c76f364
1 parent e300c82
commit c76f364
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/dist_config.py b/dist_config.py
@@ -89,7 +89,7 @@
         'platform_version': '12.x',
         # Use the latest CUDA version for build.
         # Note: CUDA 12 still supports RHEL/CentOS 7 on x86_64
-        'image': 'cupy/cupy-release-tools:cuda-runfile-12.5.0-centos7',
+        'image': 'cupy/cupy-release-tools:cuda-runfile-12.6.0-centos7',
         'libs': [],
         'includes': [],
         'preloads': ['cutensor', 'nccl', 'cudnn'],
@@ -102,6 +102,7 @@
             '12.3.0-runtime-ubuntu22.04',
             '12.4.0-runtime-ubuntu22.04',
             '12.5.0-runtime-ubuntu22.04',
+            '12.6.0-runtime-ubuntu22.04',
         ],
         'system_packages': '',
     },
@@ -112,7 +113,7 @@
         'arch': 'aarch64',
         'platform_version': '12.x',
         # Use the latest image.
-        'image': 'cupy/cupy-release-tools:cuda-runfile-12.5.0-el8',
+        'image': 'cupy/cupy-release-tools:cuda-runfile-12.6.0-el8',
         'libs': [],
         'includes': [],
         'preloads': [],
@@ -126,6 +127,7 @@
             '12.3.0-runtime-ubi8',
             '12.4.0-runtime-ubi8',
             '12.5.0-runtime-ubi8',
+            '12.6.0-runtime-ubi8',
         ],
         'system_packages': '',
     },
@@ -178,7 +180,7 @@
         'libs': [],
         'preloads': ['cutensor', 'cudnn'],
         'cudart_lib': 'cudart64_12',  # binary compatible between CUDA 12.x
-        'check_version': lambda x: 12050 <= x < 12060,  # CUDA 12.5
+        'check_version': lambda x: 12060 <= x < 12070,  # CUDA 12.6
     }
 }