b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

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

Browse source 
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
parent 1f9ab74d4c 92041430d4
commit 8c7203e105
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
pkgs/development/python-modules/pytorch/bin.nix
View file

@ -11,17 +11,20 @@
, patchelf
, pyyaml
, requests
, typing-extensions
}:
let
pyVerNoDot = builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion;
platform = if stdenv.isDarwin then "darwin" else "linux";
srcs = import ./binary-hashes.nix;
srcs = import ./binary-hashes.nix version;
unsupported = throw "Unsupported system";
version = "1.7.0";
in buildPythonPackage {
inherit version;
pname = "pytorch";
# Don't forget to update pytorch to the same version.
version = "1.6.0";
format = "wheel";
@ -39,6 +42,17 @@ in buildPythonPackage {
numpy
pyyaml
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 = ''

10
pkgs/development/python-modules/pytorch/binary-hashes.nix
View file

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