diff options
author | Jean-Philippe Bernardy <jeanphilippe.bernardy@gmail.com> | 2018-05-31 09:12:59 +0200 |
---|---|---|
committer | Jean-Philippe Bernardy <jeanphilippe.bernardy@gmail.com> | 2018-05-31 22:21:53 +0200 |
commit | 9e730ee2543dae09d8121a35b148689ceee207a4 (patch) | |
tree | 7744a370d3a0065024185c2054780e817be65723 /pkgs/top-level/python-packages.nix | |
parent | 1f14408b0e499dcc5faf656c1d7c8b21b520a72c (diff) | |
download | nixlib-9e730ee2543dae09d8121a35b148689ceee207a4.tar nixlib-9e730ee2543dae09d8121a35b148689ceee207a4.tar.gz nixlib-9e730ee2543dae09d8121a35b148689ceee207a4.tar.bz2 nixlib-9e730ee2543dae09d8121a35b148689ceee207a4.tar.lz nixlib-9e730ee2543dae09d8121a35b148689ceee207a4.tar.xz nixlib-9e730ee2543dae09d8121a35b148689ceee207a4.tar.zst nixlib-9e730ee2543dae09d8121a35b148689ceee207a4.zip |
renaming cudatoolkit version numbers
Diffstat (limited to 'pkgs/top-level/python-packages.nix')
-rw-r--r-- | pkgs/top-level/python-packages.nix | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index b5849a0bc714..6bf9f27291a8 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -1199,8 +1199,8 @@ in { cufflinks = callPackage ../development/python-modules/cufflinks { }; cupy = callPackage ../development/python-modules/cupy { - cudatoolkit = pkgs.cudatoolkit8; - cudnn = pkgs.cudnn6_cudatoolkit8; + cudatoolkit = pkgs.cudatoolkit_8; + cudnn = pkgs.cudnn6_cudatoolkit_8; nccl = pkgs.nccl; }; @@ -3968,7 +3968,7 @@ in { }; pycuda = callPackage ../development/python-modules/pycuda rec { - cudatoolkit = pkgs.cudatoolkit75; + cudatoolkit = pkgs.cudatoolkit_7_5; inherit (pkgs.stdenv) mkDerivation; }; @@ -5600,14 +5600,14 @@ in { # 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 { + cudatoolkit_9 = pkgs.cudatoolkit_9.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; + cudatoolkit = cudatoolkit_9; + cudnn = pkgs.cudnn_cudatoolkit_9.override { + inherit cudatoolkit_9; }; }; @@ -17583,8 +17583,8 @@ EOF else callPackage ../development/python-modules/tensorflow rec { cudaSupport = pkgs.config.cudaSupport or false; inherit (pkgs.linuxPackages) nvidia_x11; - cudatoolkit = pkgs.cudatoolkit9; - cudnn = pkgs.cudnn_cudatoolkit9; + cudatoolkit = pkgs.cudatoolkit_9; + cudnn = pkgs.cudnn_cudatoolkit_9; }; tensorflowWithoutCuda = self.tensorflow.override { |