Skip to content

Commit 33417c2

Browse files
lassoanpmeier
andauthored
Add cuda 11.7 and 11.8 to minimum driver version list (#108)
Co-authored-by: Philip Meier <[email protected]>
1 parent 095af73 commit 33417c2

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

light_the_torch/_cb.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,9 @@ def _detect_nvidia_driver_version() -> Optional[Version]:
150150
# Table 3 from https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
151151
_MINIMUM_DRIVER_VERSIONS = {
152152
"Linux": {
153-
Version("11.6"): Version("510.39.01"),
153+
Version("11.8"): Version("520.61.05"),
154+
Version("11.7"): Version("515.48.07"),
155+
Version("11.6"): Version("510.47.03"),
154156
Version("11.5"): Version("495.29.05"),
155157
Version("11.4"): Version("470.82.01"),
156158
Version("11.3"): Version("465.19.01"),
@@ -166,7 +168,9 @@ def _detect_nvidia_driver_version() -> Optional[Version]:
166168
Version("8.0"): Version("375.26"),
167169
},
168170
"Windows": {
169-
Version("11.6"): Version("511.23"),
171+
Version("11.8"): Version("520.06"),
172+
Version("11.7"): Version("516.31"),
173+
Version("11.6"): Version("511.65"),
170174
Version("11.5"): Version("496.13"),
171175
Version("11.4"): Version("472.50"),
172176
Version("11.3"): Version("465.89"),

0 commit comments

Comments
 (0)