From 9327db355462f78f7c85fe8d40851ebea8266c2f Mon Sep 17 00:00:00 2001 From: Thomas Watson Date: Fri, 26 Nov 2021 02:28:16 +0000 Subject: [PATCH] python3Packages.tensorflowWithCuda: upgrade cuda to recommended 11.2 to fix runtime warning --- pkgs/top-level/python-packages.nix | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 6b883ad139d2..43cca608b0b2 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -9225,15 +9225,15 @@ in { tensorflow-bin = callPackage ../development/python-modules/tensorflow/bin.nix { cudaSupport = pkgs.config.cudaSupport or false; - cudatoolkit = pkgs.cudatoolkit_11_0; - cudnn = pkgs.cudnn_cudatoolkit_11_0; + cudatoolkit = pkgs.cudatoolkit_11_2; + cudnn = pkgs.cudnn_cudatoolkit_11_2; }; tensorflow-build = callPackage ../development/python-modules/tensorflow { inherit (pkgs.darwin) cctools; cudaSupport = pkgs.config.cudaSupport or false; - cudatoolkit = pkgs.cudatoolkit_11_0; - cudnn = pkgs.cudnn_cudatoolkit_11_0; + cudatoolkit = pkgs.cudatoolkit_11_2; + cudnn = pkgs.cudnn_cudatoolkit_11_2; nccl = pkgs.nccl_cudatoolkit_11; inherit (pkgs.darwin.apple_sdk.frameworks) Foundation Security; flatbuffers-core = pkgs.flatbuffers;