python.pkgs.tensorflow: fix build with cudatoolkit 10.2

2020-01-04 17:34:29 +01:00
parent e7e5fc9584
commit 9a982d3540
2 changed files with 14 additions and 0 deletions
--- a/pkgs/development/python-modules/tensorflow/default.nix
+++ b/pkgs/development/python-modules/tensorflow/default.nix
@@ -124,6 +124,10 @@ let
        sha256 = "077cpj0kzyqxzdya1dwh8df17zfzhqn7c685hx6iskvw2979zg2n";
      })
      ./lift-gast-restriction.patch
+
+      # cuda 10.2 does not have "-bin2c-path" option anymore
+      # https://github.com/tensorflow/tensorflow/issues/34429
+      ./cuda-10.2-no-bin2c-path.patch
    ];

    # On update, it can be useful to steal the changes from gentoo