Commit graph

20 commits

Author SHA1 Message Date
Anders Kaseorg 27d13628ee pytorch: Move cudatoolkit to nativeBuildInputs
nvcc must be available in PATH at build time; otherwise CUDA support
will be disabled.

Signed-off-by: Anders Kaseorg <andersk@mit.edu>
2019-04-27 07:49:42 +02:00
Tom Hunger 488230c253 pytorch: Use nativeBuildInputs to specify binary build dependencies. 2019-03-07 09:41:43 +01:00
Austin Seipp af53114a09 pythonPackages.pytorch: fix Intel MKL BLAS compile failure
Signed-off-by: Austin Seipp <aseipp@pobox.com>
2018-12-25 10:49:40 -06:00
Austin Seipp 60176e005d pythonPackages.pytorch: add libnuma dep on Linux
Signed-off-by: Austin Seipp <aseipp@pobox.com>
2018-12-25 10:49:40 -06:00
Austin Seipp 79d21285dd pythonPackages.pytorch: meta touch up, add myself as maintainer
Signed-off-by: Austin Seipp <aseipp@pobox.com>
2018-12-25 10:49:39 -06:00
Austin Seipp 6fcebb6427 pythonPackages.pytorch: fix spurious test failures
This was fixed upstream already -- these tests ensure multi-GPU
functionality works, and thus requires multiple GPUs.

Signed-off-by: Austin Seipp <aseipp@pobox.com>
2018-12-25 10:49:39 -06:00
Austin Seipp c6ecb3887f pythonPackages.pytorch: fix weirdly broken wheel version number
Otherwise, the wheel gets built with invalid metadata -- causing
'torch >= 1.0.0' to be unsatisfiable in other python packages, for
instance.

Signed-off-by: Austin Seipp <aseipp@pobox.com>
2018-12-25 10:49:39 -06:00
Kai Wohlfahrt be96560327 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>
2018-12-25 10:49:39 -06:00
Kai Wohlfahrt 4d8cf0b84e pytorch: disable distributed tests
These fail with sandboxing
2018-11-07 17:09:54 +00:00
Dmitry Vyal f7f8b60ace pytorch-0.4 rpath fix, some tests reenabled 2018-09-23 09:36:32 +02:00
teh 8c3514609b pytorch: 0.4.0 -> 0.4.1 (#45773)
Also disable tests that fail in sandbox.
2018-08-30 17:30:29 +02:00
volth 52f53c69ce pkgs/*: remove unreferenced function arguments 2018-07-21 02:48:04 +00:00
Vladimír Čunát 2059bf9e39
Merge older staging with added firefox-related updates 2018-06-29 21:06:39 +02:00
Ryan Mulligan fce8f26af6 treewide: http -> https (#42665) 2018-06-27 22:12:57 +02:00
Frederik Rietdijk ced21f5e1a pythonPackages: remove name attribute`
The `buildPython*` function computes name from `pname` and `version`.
This change removes `name` attribute from all expressions in
`pkgs/development/python-modules`.

While at it, some other minor changes were made as well, such as
replacing `fetchurl` calls with `fetchPypi`.
2018-06-23 18:14:26 +02:00
Anders Kaseorg d2f7acf30c pytorch: 0.3.1 -> 0.4.0
Signed-off-by: Anders Kaseorg <andersk@mit.edu>
2018-05-13 14:02:08 -04:00
Anders Kaseorg ce00943916 pytorch: 0.2.0 → 0.3.1 with CUDA and cuDNN (#38530)
* pytorch-0.3 with optional cuda and cudnn

* pytorch tests reenabled if compiling without cuda

* pytorch: Conditionalize cudnn dependency on cudaSupport

Signed-off-by: Anders Kaseorg <andersk@mit.edu>

* pytorch: Compile with the same GCC version used by CUDA if cudaSupport

Fixes this error:

In file included from /nix/store/gv7w3c71jg627cpcff04yi6kwzpzjyap-cudatoolkit-9.1.85.1/include/host_config.h:50:0,
                 from /nix/store/gv7w3c71jg627cpcff04yi6kwzpzjyap-cudatoolkit-9.1.85.1/include/cuda_runtime.h:78,
                 from <command-line>:0:
/nix/store/gv7w3c71jg627cpcff04yi6kwzpzjyap-cudatoolkit-9.1.85.1/include/crt/host_config.h:121:2: error: #error -- unsupported GNU version! gcc versions later than 6 are not supported!
 #error -- unsupported GNU version! gcc versions later than 6 are not supported!
  ^~~~~

Signed-off-by: Anders Kaseorg <andersk@mit.edu>

* pytorch: Build with joined cudatoolkit

Similar to #30058 for TensorFlow.

Signed-off-by: Anders Kaseorg <andersk@mit.edu>

* pytorch: 0.3.0 -> 0.3.1

Signed-off-by: Anders Kaseorg <andersk@mit.edu>

* pytorch: Patch for “refcounted file mapping not supported” failure

Signed-off-by: Anders Kaseorg <andersk@mit.edu>

* pytorch: Skip distributed tests

Signed-off-by: Anders Kaseorg <andersk@mit.edu>

* pytorch: Use the stub libcuda.so from cudatoolkit for running tests

Signed-off-by: Anders Kaseorg <andersk@mit.edu>
2018-05-04 16:19:31 +02:00
Ismaïl Senhaji 98c84e42d4 pytorch: fix sha256 hash 2017-11-22 23:11:52 +01:00
Eric Sagnes dd809b930e pytorch: 0.1.12 -> 0.2.0 2017-08-19 10:22:23 +09:00
Tom Hunger 567b84dd50 python.pkgs.pytorch: init at 0.1.12 2017-07-17 14:08:15 +02:00