diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix index 00e4c93d9c80..00e28d65be54 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"