diff options
author | Kai Wohlfahrt <kai@prodo.ai> | 2018-12-17 14:15:13 +0000 |
---|---|---|
committer | Austin Seipp <aseipp@pobox.com> | 2018-12-25 10:49:39 -0600 |
commit | be96560327748a6f1b9fc7201df522449f3f765c (patch) | |
tree | e65a9544809046b226ac354c2705e445ebdb3622 /pkgs | |
parent | 1e1219b5909bd34d4ed00c0d2169ebdfb9b3335d (diff) | |
download | nixlib-be96560327748a6f1b9fc7201df522449f3f765c.tar nixlib-be96560327748a6f1b9fc7201df522449f3f765c.tar.gz nixlib-be96560327748a6f1b9fc7201df522449f3f765c.tar.bz2 nixlib-be96560327748a6f1b9fc7201df522449f3f765c.tar.lz nixlib-be96560327748a6f1b9fc7201df522449f3f765c.tar.xz nixlib-be96560327748a6f1b9fc7201df522449f3f765c.tar.zst nixlib-be96560327748a6f1b9fc7201df522449f3f765c.zip |
pythonPackages.pytorch: 0.4.1 -> 1.0.0
`thd_distributed` is broken just like `distributed` is. `cpp_extensions` is broken upstream now, it seems? In the future it can hopefully be re-enabled. Signed-off-by: Austin Seipp <aseipp@pobox.com>
Diffstat (limited to 'pkgs')
-rw-r--r-- | pkgs/development/python-modules/pytorch/default.nix | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix index b0ae9ef0cf55..d0aa2e59ffca 100644 --- a/pkgs/development/python-modules/pytorch/default.nix +++ b/pkgs/development/python-modules/pytorch/default.nix @@ -1,6 +1,6 @@ { buildPythonPackage, pythonOlder, cudaSupport ? false, cudatoolkit ? null, cudnn ? null, - fetchFromGitHub, lib, numpy, pyyaml, cffi, typing, cmake, + fetchFromGitHub, lib, numpy, pyyaml, cffi, typing, cmake, hypothesis, linkFarm, symlinkJoin, utillinux, which }: @@ -25,7 +25,7 @@ let "LD_LIBRARY_PATH=${cudaStub}\${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH} "; in buildPythonPackage rec { - version = "0.4.1"; + version = "1.0.0"; pname = "pytorch"; src = fetchFromGitHub { @@ -33,7 +33,7 @@ in buildPythonPackage rec { repo = "pytorch"; rev = "v${version}"; fetchSubmodules = true; - sha256 = "1cr8h47jxgfar5bamyvlayvqymnb2qvp7rr0ka2d2d4rdldf9lrp"; + sha256 = "076cpbig4sywn9vv674c0xdg832sdrd5pk1d0725pjkm436kpvlm"; }; preConfigure = lib.optionalString cudaSupport '' @@ -71,8 +71,9 @@ in buildPythonPackage rec { pyyaml ] ++ lib.optional (pythonOlder "3.5") typing; + checkInputs = [ hypothesis ]; checkPhase = '' - ${cudaStubEnv}python test/run_test.py --exclude dataloader sparse torch utils distributed + ${cudaStubEnv}python test/run_test.py --exclude dataloader sparse torch utils thd_distributed distributed cpp_extensions ''; meta = { |