nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-09-27 18:30:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

tensorflow: Add contrib/tensor_forest to rpath

Browse Source 
(cherry picked from commit 69af83b202)
This commit is contained in:
Herman Fries
2018-09-21 11:36:43 +02:00
committed by Roger Qiu
parent 38a99219f3
commit a8511a6f86
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pkgs/development/python-modules/tensorflow/bin.nix
View File

@@ -72,7 +72,7 @@ in buildPythonPackage rec {
([ stdenv.cc.cc.lib zlib ] ++ lib.optionals cudaSupport [ cudatoolkit_joined cudnn nvidia_x11 ]);
in
lib.optionalString (stdenv.isLinux) ''
rrPath="$out/${python.sitePackages}/tensorflow/:${rpath}"
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" {} \;
'';