From 427a7c0421c9ae7a6c696ee8ff39228834a99f4a Mon Sep 17 00:00:00 2001 From: Bryan Honof Date: Thu, 12 Mar 2026 11:49:45 +0100 Subject: [PATCH] python3Packages.torch: sync supportedTorchCudaCapabilities Looks like these weren't synced since a while? --- pkgs/development/python-modules/torch/source/default.nix | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix index ed985066ba62..41315bb018f0 100644 --- a/pkgs/development/python-modules/torch/source/default.nix +++ b/pkgs/development/python-modules/torch/source/default.nix @@ -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"