Merge pull request #101927 from danieldk/pytorch-bin-1.7.0

python3Packages.pytorch-bin: 1.6.0 -> 1.7.0
This commit is contained in:
Martin Weinelt 2020-10-28 18:05:07 +01:00 committed by GitHub
commit 8c7203e105
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 7 deletions

View File

@ -11,17 +11,20 @@
, patchelf , patchelf
, pyyaml , pyyaml
, requests , requests
, typing-extensions
}: }:
let let
pyVerNoDot = builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion; pyVerNoDot = builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion;
platform = if stdenv.isDarwin then "darwin" else "linux"; platform = if stdenv.isDarwin then "darwin" else "linux";
srcs = import ./binary-hashes.nix; srcs = import ./binary-hashes.nix version;
unsupported = throw "Unsupported system"; unsupported = throw "Unsupported system";
version = "1.7.0";
in buildPythonPackage { in buildPythonPackage {
inherit version;
pname = "pytorch"; pname = "pytorch";
# Don't forget to update pytorch to the same version. # Don't forget to update pytorch to the same version.
version = "1.6.0";
format = "wheel"; format = "wheel";
@ -39,6 +42,17 @@ in buildPythonPackage {
numpy numpy
pyyaml pyyaml
requests requests
typing-extensions
];
# PyTorch are broken: the dataclasses wheel is required, but ships with
# Python >= 3.7. Our dataclasses derivation is incompatible with >= 3.7.
#
# https://github.com/pytorch/pytorch/issues/46930
#
# Should be removed with the next PyTorch version.
pipInstallFlags = [
"--no-deps"
]; ];
postInstall = '' postInstall = ''

View File

@ -1,10 +1,10 @@
{ version: {
x86_64-linux-37 = { x86_64-linux-37 = {
url = "https://download.pytorch.org/whl/cu102/torch-1.6.0-cp37-cp37m-linux_x86_64.whl"; url = "https://download.pytorch.org/whl/cu102/torch-${version}-cp37-cp37m-linux_x86_64.whl";
sha256 = "0xhwv68j8gvahfzcp43bqp2x71iwv6zjhkw2f1hb82xps40mrml7"; sha256 = "0cvafwzd44d7i2d0y01lyk75lhxffcf6c94cx0wla0cnhcg9mqlf";
}; };
x86_64-linux-38 = { x86_64-linux-38 = {
url = "https://download.pytorch.org/whl/cu102/torch-1.6.0-cp38-cp38-linux_x86_64.whl"; url = "https://download.pytorch.org/whl/cu102/torch-${version}-cp38-cp38-linux_x86_64.whl";
sha256 = "05m2l04wqzw5xvjam6zwvlmc3979cksl3hrdqc2aikrv4hz8fmsk"; sha256 = "1vxswkb2gzp87sbb5ycvyvs8aw1szp3v2mwdn4mp47yarlnbx8r6";
}; };
} }