add support for CUDA 12.1 through CUDA 12.6

Author: jamesobutler

light_the_torch/_cb.py

@@ -151,6 +151,12 @@ def _detect_nvidia_driver_version() -> Optional[Version]:
     "Linux": {
         # Table 2 from
         # https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
+        Version("12.6"): Version("525.60.13"),
+        Version("12.5"): Version("525.60.13"),
+        Version("12.4"): Version("525.60.13"),
+        Version("12.3"): Version("525.60.13"),
+        Version("12.2"): Version("525.60.13"),
+        Version("12.1"): Version("525.60.13"),
         Version("12.0"): Version("525.60.13"),
         # Table 2 from
         # https://docs.nvidia.com/cuda/archive/11.8.0/cuda-toolkit-release-notes/index.html
@@ -176,7 +182,13 @@ def _detect_nvidia_driver_version() -> Optional[Version]:
     "Windows": {
         # Table 2 from
         # https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
-        Version("12.0"): Version("527.41"),
+        Version("12.6"): Version("528.33"),
+        Version("12.5"): Version("528.33"),
+        Version("12.4"): Version("528.33"),
+        Version("12.3"): Version("528.33"),
+        Version("12.2"): Version("528.33"),
+        Version("12.1"): Version("528.33"),
+        Version("12.0"): Version("528.33"),
         # Table 2 from
         # https://docs.nvidia.com/cuda/archive/11.8.0/cuda-toolkit-release-notes/index.html
         Version("11.8"): Version("452.39"),