Add cuda 11.7 and 11.8 to minimum driver version list (#108)

Co-authored-by: Philip Meier <github.pmeier@posteo.de>
pmeier · Dec 23, 2022 · 33417c2 · 33417c2
1 parent 095af73
commit 33417c2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/light_the_torch/_cb.py b/light_the_torch/_cb.py
@@ -150,7 +150,9 @@ def _detect_nvidia_driver_version() -> Optional[Version]:
 # Table 3 from https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
 _MINIMUM_DRIVER_VERSIONS = {
     "Linux": {
-        Version("11.6"): Version("510.39.01"),
+        Version("11.8"): Version("520.61.05"),
+        Version("11.7"): Version("515.48.07"),
+        Version("11.6"): Version("510.47.03"),
         Version("11.5"): Version("495.29.05"),
         Version("11.4"): Version("470.82.01"),
         Version("11.3"): Version("465.19.01"),
@@ -166,7 +168,9 @@ def _detect_nvidia_driver_version() -> Optional[Version]:
         Version("8.0"): Version("375.26"),
     },
     "Windows": {
-        Version("11.6"): Version("511.23"),
+        Version("11.8"): Version("520.06"),
+        Version("11.7"): Version("516.31"),
+        Version("11.6"): Version("511.65"),
         Version("11.5"): Version("496.13"),
         Version("11.4"): Version("472.50"),
         Version("11.3"): Version("465.89"),