python3Packages.tensorflow: use cudatoolkit 11

2020-08-19 13:34:18 -05:00
parent 9cd240c14a
commit 59eecacd2d
2 changed files with 4 additions and 3 deletions
--- a/pkgs/development/python-modules/tensorflow/2/default.nix
+++ b/pkgs/development/python-modules/tensorflow/2/default.nix
@@ -46,6 +46,7 @@ let
    paths = [
      cudatoolkit.lib
      cudatoolkit.out
+    ] ++ lib.optionals (lib.versionOlder cudatoolkit.version "11") [
      # for some reason some of the required libs are in the targets/x86_64-linux
      # directory; not sure why but this works around it
      "${cudatoolkit}/targets/${stdenv.system}"