Unverified Commit a5863443 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.torch sync supportedTorchCudaCapabilities (#499216)

parents 941dfcd1 427a7c04

pkgs/development/python-modules/torch/source/default.nix

+3 −4
Original line number Diff line number Diff line
@@ -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"