versions.nix (108 lines of code) (raw):

[ { torchVersion = "2.6"; cudaVersion = "11.8"; cxx11Abi = false; systems = [ "x86_64-linux" ]; upstreamVariant = true; } { torchVersion = "2.6"; cudaVersion = "11.8"; cxx11Abi = true; systems = [ "x86_64-linux" ]; upstreamVariant = true; } { torchVersion = "2.6"; cudaVersion = "12.4"; cxx11Abi = false; systems = [ "x86_64-linux" ]; upstreamVariant = true; } { torchVersion = "2.6"; cudaVersion = "12.4"; cxx11Abi = true; systems = [ "x86_64-linux" ]; upstreamVariant = true; } { torchVersion = "2.6"; cudaVersion = "12.6"; cxx11Abi = false; systems = [ "x86_64-linux" "aarch64-linux" ]; upstreamVariant = true; } { torchVersion = "2.6"; cudaVersion = "12.6"; cxx11Abi = true; systems = [ "x86_64-linux" "aarch64-linux" ]; upstreamVariant = true; } { torchVersion = "2.6"; rocmVersion = "6.2.4"; cxx11Abi = true; systems = [ "x86_64-linux" ]; upstreamVariant = true; } { torchVersion = "2.7"; cudaVersion = "11.8"; cxx11Abi = true; systems = [ "x86_64-linux" ]; upstreamVariant = true; } { torchVersion = "2.7"; cudaVersion = "12.6"; cxx11Abi = true; systems = [ "x86_64-linux" "aarch64-linux" ]; upstreamVariant = true; } { torchVersion = "2.7"; cudaVersion = "12.8"; cxx11Abi = true; systems = [ "x86_64-linux" "aarch64-linux" ]; upstreamVariant = true; } { torchVersion = "2.7"; rocmVersion = "6.3.4"; cxx11Abi = true; systems = [ "x86_64-linux" ]; upstreamVariant = true; } { torchVersion = "2.7"; cxx11Abi = true; metal = true; systems = [ "aarch64-darwin" ]; upstreamVariant = true; } # Non-standard versions; not included in bundle builds. { torchVersion = "2.7"; cudaVersion = "12.9"; cxx11Abi = true; systems = [ "x86_64-linux" "aarch64-linux" ]; } ]