about summary refs log tree commit diff
path: root/pkgs/top-level/python-packages.nix
diff options
context:
space:
mode:
authorJean-Philippe Bernardy <jeanphilippe.bernardy@gmail.com>2018-05-30 16:38:51 +0200
committerJean-Philippe Bernardy <jeanphilippe.bernardy@gmail.com>2018-05-31 22:21:54 +0200
commit82540b5e1a1cf56638f21326999d271eb85c9499 (patch)
treea9577b7a36d1a63cce507d2121cb0d4ca3c6ac2e /pkgs/top-level/python-packages.nix
parent397c4b7748df276e6b1b891998785b71999f5e95 (diff)
downloadnixlib-82540b5e1a1cf56638f21326999d271eb85c9499.tar
nixlib-82540b5e1a1cf56638f21326999d271eb85c9499.tar.gz
nixlib-82540b5e1a1cf56638f21326999d271eb85c9499.tar.bz2
nixlib-82540b5e1a1cf56638f21326999d271eb85c9499.tar.lz
nixlib-82540b5e1a1cf56638f21326999d271eb85c9499.tar.xz
nixlib-82540b5e1a1cf56638f21326999d271eb85c9499.tar.zst
nixlib-82540b5e1a1cf56638f21326999d271eb85c9499.zip
pythonPackages.tensorflow: 1.5.0 -> 1.7.1
Re-instates binary build for all versions.
Diffstat (limited to 'pkgs/top-level/python-packages.nix')
-rw-r--r--pkgs/top-level/python-packages.nix6
1 files changed, 3 insertions, 3 deletions
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index 6bf9f27291a8..b6c383bfc997 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -17580,11 +17580,11 @@ EOF
   tensorflow =
     if stdenv.isDarwin
     then callPackage ../development/python-modules/tensorflow/bin.nix { }
-    else callPackage ../development/python-modules/tensorflow rec {
+    else callPackage ../development/python-modules/tensorflow/bin.nix rec {
       cudaSupport = pkgs.config.cudaSupport or false;
       inherit (pkgs.linuxPackages) nvidia_x11;
-      cudatoolkit = pkgs.cudatoolkit_9;
-      cudnn = pkgs.cudnn_cudatoolkit_9;
+      cudatoolkit = pkgs.cudatoolkit_9_0;
+      cudnn = pkgs.cudnn_cudatoolkit_9_0;
     };
 
   tensorflowWithoutCuda = self.tensorflow.override {