diff options
author | Anders Kaseorg <andersk@mit.edu> | 2018-05-04 18:00:50 -0400 |
---|---|---|
committer | Anders Kaseorg <andersk@mit.edu> | 2018-05-13 14:02:08 -0400 |
commit | d2f7acf30c6dc1fe3a742f4035b11fe57514952b (patch) | |
tree | 7aae3d07263bc5074bd1fec41d29285827a64792 /pkgs/top-level | |
parent | 6db7f92cc2af827e8b8b181bf5ed828a1d0f141d (diff) | |
download | nixpkgs-d2f7acf30c6dc1fe3a742f4035b11fe57514952b.tar nixpkgs-d2f7acf30c6dc1fe3a742f4035b11fe57514952b.tar.gz nixpkgs-d2f7acf30c6dc1fe3a742f4035b11fe57514952b.tar.bz2 nixpkgs-d2f7acf30c6dc1fe3a742f4035b11fe57514952b.tar.lz nixpkgs-d2f7acf30c6dc1fe3a742f4035b11fe57514952b.tar.xz nixpkgs-d2f7acf30c6dc1fe3a742f4035b11fe57514952b.tar.zst nixpkgs-d2f7acf30c6dc1fe3a742f4035b11fe57514952b.zip |
pytorch: 0.3.1 -> 0.4.0
Signed-off-by: Anders Kaseorg <andersk@mit.edu>
Diffstat (limited to 'pkgs/top-level')
-rw-r--r-- | pkgs/top-level/python-packages.nix | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 7868f7d7ff0..95b7f8a5718 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -5623,8 +5623,20 @@ in { }; }; - pytorch = callPackage ../development/python-modules/pytorch { + pytorch = let + # Fails with CUDA 9.1 and GCC 6.4: + # https://github.com/pytorch/pytorch/issues/5831 + # https://devtalk.nvidia.com/default/topic/1028112 + # We should be able to remove this when CUDA 9.2 is released. + cudatoolkit9 = pkgs.cudatoolkit9.override { + gcc6 = pkgs.gcc5; + }; + in callPackage ../development/python-modules/pytorch { cudaSupport = pkgs.config.cudaSupport or false; + cudatoolkit = cudatoolkit9; + cudnn = pkgs.cudnn_cudatoolkit9.override { + inherit cudatoolkit9; + }; }; pytorchWithCuda = self.pytorch.override { |