зеркало из https://github.com/microsoft/DeepSpeed.git
Add CUDA 12.8 support and comment on CUDA 12.7 (#6975)
This commit is contained in:
Родитель
1640f6df4f
Коммит
46c6c9e6ce
|
@ -76,7 +76,8 @@ def get_default_compute_capabilities():
|
|||
cuda_minor_mismatch_ok = {
|
||||
10: ["10.0", "10.1", "10.2"],
|
||||
11: ["11.0", "11.1", "11.2", "11.3", "11.4", "11.5", "11.6", "11.7", "11.8"],
|
||||
12: ["12.0", "12.1", "12.2", "12.3", "12.4", "12.5", "12.6"],
|
||||
12: ["12.0", "12.1", "12.2", "12.3", "12.4", "12.5", "12.6",
|
||||
"12.8"], # There does not appear to be a CUDA Toolkit 12.7
|
||||
}
|
||||
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче