python3Packages.torch sync supportedTorchCudaCapabilities (#499216)

2026-03-22 11:43:54 +00:00
parent 941dfcd183 427a7c0421
commit a586344345
1 changed files with 3 additions and 4 deletions
@@ -118,7 +118,7 @@ let

  setBool = v: if v then "1" else "0";

-  # https://github.com/pytorch/pytorch/blob/v2.8.0/torch/utils/cpp_extension.py#L2411-L2414
+  # https://github.com/pytorch/pytorch/blob/v2.10.0/torch/utils/cpp_extension.py#L2411-L2414
  supportedTorchCudaCapabilities =
    let
      real = [
@@ -140,10 +140,9 @@ let
        "9.0"
        "9.0a"
        "10.0"
-        "10.0"
        "10.0a"
-        "10.1"
-        "10.1a"
+        "11.0"
+        "11.0a"
        "10.3"
        "10.3a"
        "12.0"