diff --git a/light_the_torch/_cb.py b/light_the_torch/_cb.py index 5bef516..9cc0ba6 100644 --- 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"),