pythonPackages.tensorflow: 1.11 -> 1.13.1
This commit is contained in:
committed by
Frederik Rietdijk
parent
d1b7ada0ba
commit
1c350dc99e
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user