about summary refs log tree commit diff
path: root/pkgs/top-level
diff options
context:
space:
mode:
authorDaniƫl de Kok <me@danieldk.eu>2020-09-30 11:37:42 +0200
committerGitHub <noreply@github.com>2020-09-30 11:37:42 +0200
commitb751c120cef6d42ed26093df5363800584c97c1f (patch)
tree7d1f8d8b64b11ebae656a4944492e9035d29f2d5 /pkgs/top-level
parentd7702e0dbab81a24c27b57d26ddb613f6d451749 (diff)
parent59eecacd2d2c0ebf32e393a2c111b38755004f53 (diff)
Merge pull request #95824 from obsidiansystems/python3.tensorflow_2-update-to-2.3.0
tensorflow: update to 2.3.0
Diffstat (limited to 'pkgs/top-level')
-rw-r--r--pkgs/top-level/python-packages.nix7
1 files changed, 3 insertions, 4 deletions
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index 26f94af1bd8ef..83df9f6a79934 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -6932,10 +6932,9 @@ in {
 
   tensorflow-build_2 = callPackage ../development/python-modules/tensorflow/2 {
     cudaSupport = pkgs.config.cudaSupport or false;
-    inherit (pkgs.linuxPackages) nvidia_x11;
-    cudatoolkit = pkgs.cudatoolkit_10;
-    cudnn = pkgs.cudnn_cudatoolkit_10;
-    nccl = pkgs.nccl_cudatoolkit_10;
+    cudatoolkit = pkgs.cudatoolkit_11;
+    cudnn = pkgs.cudnn_cudatoolkit_11;
+    nccl = pkgs.nccl_cudatoolkit_11;
     openssl = pkgs.openssl_1_1;
     inherit (pkgs.darwin.apple_sdk.frameworks) Foundation Security;
   };