From 450a4bf8a46c61505dcfa2fc4543074f50b22c69 Mon Sep 17 00:00:00 2001 From: Gaetan Lepage Date: Fri, 25 Apr 2025 08:48:20 +0200 Subject: [PATCH] python312Packages.torch-bin: 2.6.0 -> 2.7.0 Diff: https://github.com/pytorch/pytorch/compare/v2.6.0...v2.7.0 Changelog: https://github.com/pytorch/pytorch/releases/tag/v2.7.0 --- pkgs/development/python-modules/torch/bin.nix | 3 +- .../python-modules/torch/binary-hashes.nix | 92 +++++++++---------- .../python-modules/torch/prefetch.sh | 12 +-- 3 files changed, 54 insertions(+), 53 deletions(-) diff --git a/pkgs/development/python-modules/torch/bin.nix b/pkgs/development/python-modules/torch/bin.nix index 26f48307621b..5ce60ba8259b 100644 --- a/pkgs/development/python-modules/torch/bin.nix +++ b/pkgs/development/python-modules/torch/bin.nix @@ -35,7 +35,7 @@ let pyVerNoDot = builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion; srcs = import ./binary-hashes.nix version; unsupported = throw "Unsupported system"; - version = "2.6.0"; + version = "2.7.0"; in buildPythonPackage { inherit version; @@ -69,6 +69,7 @@ buildPythonPackage { cusparselt libcublas libcufft + libcufile libcurand libcusolver libcusparse diff --git a/pkgs/development/python-modules/torch/binary-hashes.nix b/pkgs/development/python-modules/torch/binary-hashes.nix index d3afcf155318..a94a39aae52a 100644 --- a/pkgs/development/python-modules/torch/binary-hashes.nix +++ b/pkgs/development/python-modules/torch/binary-hashes.nix @@ -7,81 +7,81 @@ version: builtins.getAttr version { - "2.6.0" = { + "2.7.0" = { x86_64-linux-39 = { - name = "torch-2.6.0-cp39-cp39-linux_x86_64.whl"; - url = "https://download.pytorch.org/whl/cu124/torch-2.6.0%2Bcu124-cp39-cp39-linux_x86_64.whl"; - hash = "sha256-5mEmfNAkJGKrEAvdZ/ZRmIqp9n6zFgnWkJr8rIkd9hI="; + name = "torch-2.7.0-cp39-cp39-linux_x86_64.whl"; + url = "https://download.pytorch.org/whl/cu128/torch-2.7.0%2Bcu128-cp39-cp39-manylinux_2_28_x86_64.whl"; + hash = "sha256-9Eb5eyDLBwdHsQP7ZA35QbiMtoyNOwFTgofQXVan6HQ="; }; x86_64-linux-310 = { - name = "torch-2.6.0-cp310-cp310-linux_x86_64.whl"; - url = "https://download.pytorch.org/whl/cu124/torch-2.6.0%2Bcu124-cp310-cp310-linux_x86_64.whl"; - hash = "sha256-fyun98BFkyClIWlva1vMwYf1mJCyPJ37bEmwuHxr/Jc="; + name = "torch-2.7.0-cp310-cp310-linux_x86_64.whl"; + url = "https://download.pytorch.org/whl/cu128/torch-2.7.0%2Bcu128-cp310-cp310-manylinux_2_28_x86_64.whl"; + hash = "sha256-rBhJVT7mc9+vtExhDGDLYKKJDw4Rf0NZmlJs93fri4w="; }; x86_64-linux-311 = { - name = "torch-2.6.0-cp311-cp311-linux_x86_64.whl"; - url = "https://download.pytorch.org/whl/cu124/torch-2.6.0%2Bcu124-cp311-cp311-linux_x86_64.whl"; - hash = "sha256-1MPpqNMafA/LudoXwxoZF+H6wmxWakz72MlWitfK3nk="; + name = "torch-2.7.0-cp311-cp311-linux_x86_64.whl"; + url = "https://download.pytorch.org/whl/cu128/torch-2.7.0%2Bcu128-cp311-cp311-manylinux_2_28_x86_64.whl"; + hash = "sha256-xLvAtL5gMZuhzvyQvpVXsxfws8Jh7s65bKbgND7sVr8="; }; x86_64-linux-312 = { - name = "torch-2.6.0-cp312-cp312-linux_x86_64.whl"; - url = "https://download.pytorch.org/whl/cu124/torch-2.6.0%2Bcu124-cp312-cp312-linux_x86_64.whl"; - hash = "sha256-o5O1BoRANcDawvMOqEeMNDuOlaQp8G87PK38f1OttZc="; + name = "torch-2.7.0-cp312-cp312-linux_x86_64.whl"; + url = "https://download.pytorch.org/whl/cu128/torch-2.7.0%2Bcu128-cp312-cp312-manylinux_2_28_x86_64.whl"; + hash = "sha256-fA8I0cRKAqutOJNz3d/OdZBLlppBC+L05RCUg909wM4="; }; x86_64-linux-313 = { - name = "torch-2.6.0-cp313-cp313-linux_x86_64.whl"; - url = "https://download.pytorch.org/whl/cu124/torch-2.6.0%2Bcu124-cp313-cp313-linux_x86_64.whl"; - hash = "sha256-DzvFPJiM6VaM2HaipTFnYehKhwQTXsgGj1+BtEF5ecs="; + name = "torch-2.7.0-cp313-cp313-linux_x86_64.whl"; + url = "https://download.pytorch.org/whl/cu128/torch-2.7.0%2Bcu128-cp313-cp313-manylinux_2_28_x86_64.whl"; + hash = "sha256-0vafkJ2l3FIRPsZqhR1iB589UsgxhM9kvuvfEsovcFw="; }; aarch64-darwin-39 = { - name = "torch-2.6.0-cp39-none-macosx_11_0_arm64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp39-none-macosx_11_0_arm64.whl"; - hash = "sha256-Jl9w3l/UW4ZNkktkvheX+G52yOSKAsKjpvx+wkfSImw="; + name = "torch-2.7.0-cp39-none-macosx_11_0_arm64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0-cp39-none-macosx_11_0_arm64.whl"; + hash = "sha256-zNdQkUFxOZeGG3qUfvCnFxQ81+kkCt3RaPOLqP0j/VY="; }; aarch64-darwin-310 = { - name = "torch-2.6.0-cp310-none-macosx_11_0_arm64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp310-none-macosx_11_0_arm64.whl"; - hash = "sha256-CeBvmUnhoFGMWwn+lSlbyWYfIZ2ey2+Yk+USPhBpZig="; + name = "torch-2.7.0-cp310-none-macosx_11_0_arm64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0-cp310-none-macosx_11_0_arm64.whl"; + hash = "sha256-NOAWjtbemRIWEtciJOWbKlioPa5kmZmQ6tpyYMXdWC0="; }; aarch64-darwin-311 = { - name = "torch-2.6.0-cp311-none-macosx_11_0_arm64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp311-none-macosx_11_0_arm64.whl"; - hash = "sha256-lPxjs7S+3TJ69YhpZVn2jCZEQOJQPMnmlUAZRz10riE="; + name = "torch-2.7.0-cp311-none-macosx_11_0_arm64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0-cp311-none-macosx_11_0_arm64.whl"; + hash = "sha256-Co1DyqNCuZhhAexf61u/HYZXC1yqAenLQmN4MRJY/d4="; }; aarch64-darwin-312 = { - name = "torch-2.6.0-cp312-none-macosx_11_0_arm64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp312-none-macosx_11_0_arm64.whl"; - hash = "sha256-mmEK/iFqhai5vJ+DZe1WFTXJPoBMKjF+9/q8xd7aCYk="; + name = "torch-2.7.0-cp312-none-macosx_11_0_arm64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0-cp312-none-macosx_11_0_arm64.whl"; + hash = "sha256-MLdoiocjmn3oPyaTM2Udjlgq//zm9ZH/8IwEb3eHKW4="; }; aarch64-darwin-313 = { - name = "torch-2.6.0-cp313-none-macosx_11_0_arm64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp313-none-macosx_11_0_arm64.whl"; - hash = "sha256-/5b0A4+K+ffsQjFxDtRUnaG9662VkjlTolBF3Pb9h+I="; + name = "torch-2.7.0-cp313-none-macosx_11_0_arm64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0-cp313-none-macosx_11_0_arm64.whl"; + hash = "sha256-J/UAe99F97t69/EdGCjVwkh+AwaQr7PYmmUf1wNqOQ4="; }; aarch64-linux-39 = { - name = "torch-2.6.0-cp39-cp39-manylinux_2_28_aarch64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp39-cp39-manylinux_2_28_aarch64.whl"; - hash = "sha256-KrnGs9bupQa9qbgqAVXpdNjvjji0F1idFEVotPpZr+E="; + name = "torch-2.7.0-cp39-cp39-manylinux_2_28_aarch64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp39-cp39-manylinux_2_28_aarch64.whl"; + hash = "sha256-fQpBBrwP4zkpX1CZAM5GIo9Fua2GRmYv5Qx9nllgw8E="; }; aarch64-linux-310 = { - name = "torch-2.6.0-cp310-cp310-manylinux_2_28_aarch64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp310-cp310-manylinux_2_28_aarch64.whl"; - hash = "sha256-kIMvTRGMVmuGUqIZasaV/B8Uz0INsntaG0HH6q8hQek="; + name = "torch-2.7.0-cp310-cp310-manylinux_2_28_aarch64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp310-cp310-manylinux_2_28_aarch64.whl"; + hash = "sha256-I4aFne5hkaJXHOFcZcPhgAjU5vF9UlbUm0Zg5UZNyug="; }; aarch64-linux-311 = { - name = "torch-2.6.0-cp311-cp311-manylinux_2_28_aarch64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp311-cp311-manylinux_2_28_aarch64.whl"; - hash = "sha256-09q5+wKU8miuwo6Kq6g06dAGuQpQ21vC/iGRqdSMYIQ="; + name = "torch-2.7.0-cp311-cp311-manylinux_2_28_aarch64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp311-cp311-manylinux_2_28_aarch64.whl"; + hash = "sha256-zlEDde15Ij2z7BRP4Uy8/8ijYaxX85Z0OX/y2Ns7LCE="; }; aarch64-linux-312 = { - name = "torch-2.6.0-cp312-cp312-manylinux_2_28_aarch64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp312-cp312-manylinux_2_28_aarch64.whl"; - hash = "sha256-MYKQ6JJDU8YbElzch2jRUghwTieed1fBE7liB0Deypg="; + name = "torch-2.7.0-cp312-cp312-manylinux_2_28_aarch64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp312-cp312-manylinux_2_28_aarch64.whl"; + hash = "sha256-qEW2872jxA9zaEfe3pXYv+yB+34RRYzSWXO6E1Qs8fY="; }; aarch64-linux-313 = { - name = "torch-2.6.0-cp313-cp313-manylinux_2_28_aarch64.whl"; - url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl"; - hash = "sha256-tefo1WGyY7WtgElzYoHNEseOUee8GpE/1AmP0OC5Y0c="; + name = "torch-2.7.0-cp313-cp313-manylinux_2_28_aarch64.whl"; + url = "https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl"; + hash = "sha256-rd+RB5OVIv+ztg0pAP7oOKd9vgmOJkPgEWT0b4YS+cA="; }; }; } diff --git a/pkgs/development/python-modules/torch/prefetch.sh b/pkgs/development/python-modules/torch/prefetch.sh index 69a879bffc7e..169d5f802ab0 100755 --- a/pkgs/development/python-modules/torch/prefetch.sh +++ b/pkgs/development/python-modules/torch/prefetch.sh @@ -5,17 +5,17 @@ set -eou pipefail version=$1 -linux_cuda_version="cu124" +linux_cuda_version="cu128" linux_cuda_bucket="https://download.pytorch.org/whl/${linux_cuda_version}" linux_cpu_bucket="https://download.pytorch.org/whl/cpu" darwin_bucket="https://download.pytorch.org/whl/cpu" url_and_key_list=( - "x86_64-linux-39 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp39-cp39-linux_x86_64.whl torch-${version}-cp39-cp39-linux_x86_64.whl" - "x86_64-linux-310 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp310-cp310-linux_x86_64.whl torch-${version}-cp310-cp310-linux_x86_64.whl" - "x86_64-linux-311 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp311-cp311-linux_x86_64.whl torch-${version}-cp311-cp311-linux_x86_64.whl" - "x86_64-linux-312 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp312-cp312-linux_x86_64.whl torch-${version}-cp312-cp312-linux_x86_64.whl" - "x86_64-linux-313 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp313-cp313-linux_x86_64.whl torch-${version}-cp313-cp313-linux_x86_64.whl" + "x86_64-linux-39 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp39-cp39-manylinux_2_28_x86_64.whl torch-${version}-cp39-cp39-linux_x86_64.whl" + "x86_64-linux-310 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp310-cp310-manylinux_2_28_x86_64.whl torch-${version}-cp310-cp310-linux_x86_64.whl" + "x86_64-linux-311 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp311-cp311-manylinux_2_28_x86_64.whl torch-${version}-cp311-cp311-linux_x86_64.whl" + "x86_64-linux-312 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp312-cp312-manylinux_2_28_x86_64.whl torch-${version}-cp312-cp312-linux_x86_64.whl" + "x86_64-linux-313 $linux_cuda_bucket/torch-${version}%2B${linux_cuda_version}-cp313-cp313-manylinux_2_28_x86_64.whl torch-${version}-cp313-cp313-linux_x86_64.whl" "aarch64-darwin-39 $darwin_bucket/torch-${version}-cp39-none-macosx_11_0_arm64.whl torch-${version}-cp39-none-macosx_11_0_arm64.whl" "aarch64-darwin-310 $darwin_bucket/torch-${version}-cp310-none-macosx_11_0_arm64.whl torch-${version}-cp310-none-macosx_11_0_arm64.whl" "aarch64-darwin-311 $darwin_bucket/torch-${version}-cp311-none-macosx_11_0_arm64.whl torch-${version}-cp311-none-macosx_11_0_arm64.whl"