diff options
author | Colin <486199+c00w@users.noreply.github.com> | 2019-07-31 06:00:12 -0400 |
---|---|---|
committer | Nikolay Amiantov <ab@fmap.me> | 2019-07-31 13:00:12 +0300 |
commit | f634a929d4d75ca431cf15008a1667e7554b7d00 (patch) | |
tree | a42c738f31f735cc506dbdfc4a5021027e4cdd0a | |
parent | dc7352b1fcb002ef73c85be9a3689e8f1947f3a3 (diff) |
pythonPackages.tensorflow: Hardcode a second search class. (#65584)
It appears without this, libcuda.so.1 is not picked up and nvidia graphics are broken
-rw-r--r-- | pkgs/development/python-modules/tensorflow/bin.nix | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/tensorflow/bin.nix b/pkgs/development/python-modules/tensorflow/bin.nix index d02a4e1b9f296..817ba09dc9d2b 100644 --- a/pkgs/development/python-modules/tensorflow/bin.nix +++ b/pkgs/development/python-modules/tensorflow/bin.nix @@ -91,7 +91,7 @@ in buildPythonPackage rec { lib.optionalString (stdenv.isLinux) '' rrPath="$out/${python.sitePackages}/tensorflow/:$out/${python.sitePackages}/tensorflow/contrib/tensor_forest/:${rpath}" internalLibPath="$out/${python.sitePackages}/tensorflow/python/_pywrap_tensorflow_internal.so" - find $out -name '*${stdenv.hostPlatform.extensions.sharedLibrary}' -exec patchelf --set-rpath "$rrPath" {} \; + find $out \( -name '*.so' -or -name '*.so.*' \) -exec patchelf --set-rpath "$rrPath" {} \; ''; |