pythonPackages.tensorflow: Hardcode a second search class. (#65584)

It appears without this, libcuda.so.1 is not picked up and nvidia
graphics are broken
This commit is contained in:
Colin 2019-07-31 06:00:12 -04:00 committed by Nikolay Amiantov
parent dc7352b1fc
commit f634a929d4

View file

@ -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" {} \;
'';