pythonPackages.tensorflow: 1.11 -> 1.13.1

This commit is contained in:
Jean-Philippe Bernardy
2019-02-28 11:59:17 +01:00
committed by Frederik Rietdijk
parent d1b7ada0ba
commit 1c350dc99e
6 changed files with 41 additions and 75 deletions

View File

@@ -5064,8 +5064,8 @@ in {
else callPackage ../development/python-modules/tensorflow/bin.nix rec {
cudaSupport = pkgs.config.cudaSupport or false;
inherit (pkgs.linuxPackages) nvidia_x11;
cudatoolkit = pkgs.cudatoolkit_9_0;
cudnn = pkgs.cudnn_cudatoolkit_9_0;
cudatoolkit = pkgs.cudatoolkit_10_0;
cudnn = pkgs.cudnn_cudatoolkit_10_0;
});
tensorflowWithoutCuda = self.tensorflow.override {