git.iko.soy/typeable/nixpkgs
on master help sign in
pytorch-bin: 1.10.0 -> 1.11.0
commitad2e516c099106ed1b64b87f0d546c1245cfd8f3 authorRehno Lindeque <errantkid@gmail.com> dateFri 18 Mar 2022, 09:24 (5 years ago) parentf4aabde
3 files changed+56−31
compare with master
unifiedsplit
pkgs/development/python-modules/pytorch/bin.nix+3−2 pkgs/development/python-modules/pytorch/binary-hashes.nix+40−25 pkgs/development/python-modules/pytorch/prefetch.sh+13−4
pkgs/development/python-modules/pytorch/bin.nix+3−2
@@ -4,6 +4,7 @@
44, isPy37
55, isPy38
66, isPy39
7+, isPy310
78, python
89, addOpenGLRunpath
910, future
@@ -19,7 +20,7 @@ let
1920 pyVerNoDot = builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion;
2021 srcs = import ./binary-hashes.nix version;
2122 unsupported = throw "Unsupported system";
22− version = "1.10.0";
23+ version = "1.11.0";
2324in buildPythonPackage {
2425 inherit version;
2526
@@ -28,7 +29,7 @@ in buildPythonPackage {
2829
2930 format = "wheel";
3031
31− disabled = !(isPy37 || isPy38 || isPy39);
32+ disabled = !(isPy37 || isPy38 || isPy39 || isPy310);
3233
3334 src = fetchurl srcs."${stdenv.system}-${pyVerNoDot}" or unsupported;
3435
pkgs/development/python-modules/pytorch/binary-hashes.nix+40−25
@@ -6,46 +6,61 @@
66# To add a new version, run "prefetch.sh 'new-version'" to paste the generated file as follows.
77
88version : builtins.getAttr version {
9− "1.10.0" = {
9+ "1.11.0" = {
1010 x86_64-linux-37 = {
11− name = "torch-1.10.0-cp37-cp37m-linux_x86_64.whl";
12− url = "https://download.pytorch.org/whl/cu113/torch-1.10.0%2Bcu113-cp37-cp37m-linux_x86_64.whl";
13− hash = "sha256-KpDbklee2HXSqgrWr1U1nj8EJqUjBWp7SbACw8xtKtg=";
11+ name = "torch-1.11.0-cp37-cp37m-linux_x86_64.whl";
12+ url = "https://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp37-cp37m-linux_x86_64.whl";
13+ hash = "sha256-9WMzRw2uo8lweLN2B+ADXMz3L8XDb9hFRuGkuNmUTys=";
1414 };
1515 x86_64-linux-38 = {
16− name = "torch-1.10.0-cp38-cp38-linux_x86_64.whl";
17− url = "https://download.pytorch.org/whl/cu113/torch-1.10.0%2Bcu113-cp38-cp38-linux_x86_64.whl";
18− hash = "sha256-zM3cMriUG9A+3in/ChzOLytRETpe4ju4uXkxasIRQYM=";
16+ name = "torch-1.11.0-cp38-cp38-linux_x86_64.whl";
17+ url = "https://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp38-cp38-linux_x86_64.whl";
18+ hash = "sha256-tqeZvbbuPZFOXmK920J21KECSMGvTy0hdzjl+e4nSFs=";
1919 };
2020 x86_64-linux-39 = {
21− name = "torch-1.10.0-cp39-cp39-linux_x86_64.whl";
22− url = "https://download.pytorch.org/whl/cu113/torch-1.10.0%2Bcu113-cp39-cp39-linux_x86_64.whl";
23− hash = "sha256-w8UJDh4b5cgDu7ZSvDoKzNH4hiXEyRfvpycNOg+wJOg=";
21+ name = "torch-1.11.0-cp39-cp39-linux_x86_64.whl";
22+ url = "https://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp39-cp39-linux_x86_64.whl";
23+ hash = "sha256-6RJrCl2VcEvuQKnQ7xy9gtjceGPkY4o3a+9wLf1lk3A=";
24+ };
25+ x86_64-linux-310 = {
26+ name = "torch-1.11.0-cp310-cp310-linux_x86_64.whl";
27+ url = "https://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp310-cp310-linux_x86_64.whl";
28+ hash = "sha256-powzZXpUYTHrm8ROKpjS+nBKr66GFGCwUbgoE4Usy0Q=";
2429 };
2530 x86_64-darwin-37 = {
26− name = "torch-1.10.0-cp37-none-macosx_10_9_x86_64.whl";
27− url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp37-none-macosx_10_9_x86_64.whl";
28− hash = "sha256-RJkFVUcIfX736KdU8JwsTxRwKXrj5UkDY9umbHVQGyE=";
31+ name = "torch-1.11.0-cp37-none-macosx_10_9_x86_64.whl";
32+ url = "https://download.pytorch.org/whl/cpu/torch-1.11.0-cp37-none-macosx_10_9_x86_64.whl";
33+ hash = "sha256-aGCx0b8LsLZ6a9R/haDkyCW1GO6hO11hAZmdu8vVvAw=";
2934 };
3035 x86_64-darwin-38 = {
31− name = "torch-1.10.0-cp38-none-macosx_10_9_x86_64.whl";
32− url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp38-none-macosx_10_9_x86_64.whl";
33− hash = "sha256-rvevti6bF0tODl4eSkLjurO4SQpmjWZvYvfUUXVZ+/I=";
36+ name = "torch-1.11.0-cp38-none-macosx_10_9_x86_64.whl";
37+ url = "https://download.pytorch.org/whl/cpu/torch-1.11.0-cp38-none-macosx_10_9_x86_64.whl";
38+ hash = "sha256-DMyFzQYiej7fgJ4seV/Vdiw9Too4tcn3RMbnz4QTYbs=";
3439 };
3540 x86_64-darwin-39 = {
36− name = "torch-1.10.0-cp39-none-macosx_10_9_x86_64.whl";
37− url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp39-none-macosx_10_9_x86_64.whl";
38− hash = "sha256-1u+HRwtE35lw6EVCVH1bp3ILuJYWYCRB31VaObEk4rw=";
41+ name = "torch-1.11.0-cp39-none-macosx_10_9_x86_64.whl";
42+ url = "https://download.pytorch.org/whl/cpu/torch-1.11.0-cp39-none-macosx_10_9_x86_64.whl";
43+ hash = "sha256-UP2b+FxXjIccKPHLCs6d/GAkQBx/OZsXT7DzcImfRFQ=";
44+ };
45+ x86_64-darwin-310 = {
46+ name = "torch-1.11.0-cp310-none-macosx_10_9_x86_64.whl";
47+ url = "https://download.pytorch.org/whl/cpu/torch-1.11.0-cp39-none-macosx_10_9_x86_64.whl";
48+ hash = "sha256-UP2b+FxXjIccKPHLCs6d/GAkQBx/OZsXT7DzcImfRFQ=";
3949 };
4050 aarch64-darwin-38 = {
41− name = "torch-1.10.0-cp38-none-macosx_11_0_arm64.whl";
42− url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp38-none-macosx_11_0_arm64.whl";
43− hash = "sha256-1hhYJ7KFeAZTzdgdd6Cf3KdqWxkNWYbVUr4qXEQs+qQ=";
51+ name = "torch-1.11.0-cp38-none-macosx_11_0_arm64.whl";
52+ url = "https://download.pytorch.org/whl/cpu/torch-1.11.0-cp38-none-macosx_11_0_arm64.whl";
53+ hash = "sha256-wVVOSddPGyw+cgLXcFa6LddGVDdYW6xkBitYD3FKROk=";
4454 };
4555 aarch64-darwin-39 = {
46− name = "torch-1.10.0-cp39-none-macosx_11_0_arm64.whl";
47− url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp39-none-macosx_11_0_arm64.whl";
48− hash = "sha256-7qZ17AHsS0oGVf0phPFmpco7kz2uatTrTlLrpwJtwXY=";
56+ name = "torch-1.11.0-cp39-none-macosx_11_0_arm64.whl";
57+ url = "https://download.pytorch.org/whl/cpu/torch-1.11.0-cp39-none-macosx_11_0_arm64.whl";
58+ hash = "sha256-DkivZq11Xw+cXyZkAopBT1fEnWrcN+d+Bv4ABNpO22E=";
59+ };
60+ aarch64-darwin-310 = {
61+ name = "torch-1.11.0-cp310-none-macosx_11_0_arm64.whl";
62+ url = "https://download.pytorch.org/whl/cpu/torch-1.11.0-cp39-none-macosx_11_0_arm64.whl";
63+ hash = "sha256-DkivZq11Xw+cXyZkAopBT1fEnWrcN+d+Bv4ABNpO22E=";
4964 };
5065 };
5166}
pkgs/development/python-modules/pytorch/prefetch.sh+13−4
@@ -5,12 +5,21 @@ set -eou pipefail
55
66version=$1
77
8−bucket="https://download.pytorch.org/whl/cu113"
8+linux_bucket="https://download.pytorch.org/whl/cu113"
9+darwin_bucket="https://download.pytorch.org/whl/cpu"
910
1011url_and_key_list=(
11− "x86_64-linux-37 $bucket/torch-${version}%2Bcu113-cp37-cp37m-linux_x86_64.whl torch-${version}-cp37-cp37m-linux_x86_64.whl"
12− "x86_64-linux-38 $bucket/torch-${version}%2Bcu113-cp38-cp38-linux_x86_64.whl torch-${version}-cp38-cp38-linux_x86_64.whl"
13− "x86_64-linux-39 $bucket/torch-${version}%2Bcu113-cp39-cp39-linux_x86_64.whl torch-${version}-cp39-cp39-linux_x86_64.whl"
12+ "x86_64-linux-37 $linux_bucket/torch-${version}%2Bcu113-cp37-cp37m-linux_x86_64.whl torch-${version}-cp37-cp37m-linux_x86_64.whl"
13+ "x86_64-linux-38 $linux_bucket/torch-${version}%2Bcu113-cp38-cp38-linux_x86_64.whl torch-${version}-cp38-cp38-linux_x86_64.whl"
14+ "x86_64-linux-39 $linux_bucket/torch-${version}%2Bcu113-cp39-cp39-linux_x86_64.whl torch-${version}-cp39-cp39-linux_x86_64.whl"
15+ "x86_64-linux-310 $linux_bucket/torch-${version}%2Bcu113-cp310-cp310-linux_x86_64.whl torch-${version}-cp310-cp310-linux_x86_64.whl"
16+ "x86_64-darwin-37 $darwin_bucket/torch-${version}-cp37-none-macosx_10_9_x86_64.whl torch-${version}-cp37-none-macosx_10_9_x86_64.whl"
17+ "x86_64-darwin-38 $darwin_bucket/torch-${version}-cp38-none-macosx_10_9_x86_64.whl torch-${version}-cp38-none-macosx_10_9_x86_64.whl"
18+ "x86_64-darwin-39 $darwin_bucket/torch-${version}-cp39-none-macosx_10_9_x86_64.whl torch-${version}-cp39-none-macosx_10_9_x86_64.whl"
19+ "x86_64-darwin-310 $darwin_bucket/torch-${version}-cp310-none-macosx_10_9_x86_64.whl torch-${version}-cp310-none-macosx_10_9_x86_64.whl"
20+ "aarch64-darwin-38 $darwin_bucket/torch-${version}-cp38-none-macosx_11_0_arm64.whl torch-${version}-cp38-none-macosx_11_0_arm64.whl"
21+ "aarch64-darwin-39 $darwin_bucket/torch-${version}-cp39-none-macosx_11_0_arm64.whl torch-${version}-cp39-none-macosx_11_0_arm64.whl"
22+ "aarch64-darwin-310 $darwin_bucket/torch-${version}-cp310-none-macosx_11_0_arm64.whl torch-${version}-cp310-none-macosx_11_0_arm64.whl"
1423)
1524
1625hashfile="binary-hashes-$version.nix"