alioth/nixpkgs
3
0
Fork 0
forked from mirrors/nixpkgs
Code Releases Activity

pythonPackages.pytorch: 0.4.1 -> 1.0.0

Browse source 
`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>
This commit is contained in:
Kai Wohlfahrt 2018-12-17 14:15:13 +00:00 committed by Austin Seipp
parent 1e1219b590
commit be96560327
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pkgs/development/python-modules/pytorch/default.nix
View file

@ -1,6 +1,6 @@
{ buildPythonPackage, pythonOlder, { buildPythonPackage, pythonOlder,
cudaSupport ? false, cudatoolkit ? null, cudnn ? null, cudaSupport ? false, cudatoolkit ? null, cudnn ? null,
fetchFromGitHub, lib, numpy, pyyaml, cffi, typing, cmake, fetchFromGitHub, lib, numpy, pyyaml, cffi, typing, cmake, hypothesis,
linkFarm, symlinkJoin, linkFarm, symlinkJoin,
utillinux, which }: utillinux, which }:
@ -25,7 +25,7 @@ let
"LD_LIBRARY_PATH=${cudaStub}\${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH} "; "LD_LIBRARY_PATH=${cudaStub}\${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH} ";
in buildPythonPackage rec { in buildPythonPackage rec {
version = "0.4.1"; version = "1.0.0";
pname = "pytorch"; pname = "pytorch";
src = fetchFromGitHub { src = fetchFromGitHub {
@ -33,7 +33,7 @@ in buildPythonPackage rec {
repo = "pytorch"; repo = "pytorch";
rev = "v${version}"; rev = "v${version}";
fetchSubmodules = true; fetchSubmodules = true;
sha256 = "1cr8h47jxgfar5bamyvlayvqymnb2qvp7rr0ka2d2d4rdldf9lrp"; sha256 = "076cpbig4sywn9vv674c0xdg832sdrd5pk1d0725pjkm436kpvlm";
}; };
preConfigure = lib.optionalString cudaSupport '' preConfigure = lib.optionalString cudaSupport ''
@ -71,8 +71,9 @@ in buildPythonPackage rec {
pyyaml pyyaml
] ++ lib.optional (pythonOlder "3.5") typing; ] ++ lib.optional (pythonOlder "3.5") typing;
checkInputs = [ hypothesis ];
checkPhase = '' 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 = { meta = {