about summary refs log tree commit diff
diff options
context:
space:
mode:
authorBas van Dijk <v.dijk.bas@gmail.com>2018-07-21 02:21:30 +0200
committerPeter Simons <simons@cryp.to>2018-07-21 16:22:13 +0200
commit1c48e89eb138c75ca8ebf95a6aa13078c03be9d5 (patch)
treec9a4457febe88390068ffc6e1046eb3ff333b95f
parentaa9b50ccf33cb6ff69f8335fd6fb4d34253bef50 (diff)
downloadnixlib-1c48e89eb138c75ca8ebf95a6aa13078c03be9d5.tar
nixlib-1c48e89eb138c75ca8ebf95a6aa13078c03be9d5.tar.gz
nixlib-1c48e89eb138c75ca8ebf95a6aa13078c03be9d5.tar.bz2
nixlib-1c48e89eb138c75ca8ebf95a6aa13078c03be9d5.tar.lz
nixlib-1c48e89eb138c75ca8ebf95a6aa13078c03be9d5.tar.xz
nixlib-1c48e89eb138c75ca8ebf95a6aa13078c03be9d5.tar.zst
nixlib-1c48e89eb138c75ca8ebf95a6aa13078c03be9d5.zip
libtensorflow: init at 1.8.0
Required for haskell.packages.ghc822.tensorflow
-rw-r--r--pkgs/development/libraries/libtensorflow/default.nix32
-rw-r--r--pkgs/top-level/all-packages.nix2
2 files changed, 34 insertions, 0 deletions
diff --git a/pkgs/development/libraries/libtensorflow/default.nix b/pkgs/development/libraries/libtensorflow/default.nix
new file mode 100644
index 000000000000..fce15b121b42
--- /dev/null
+++ b/pkgs/development/libraries/libtensorflow/default.nix
@@ -0,0 +1,32 @@
+{ stdenv, fetchurl, patchelf }:
+stdenv.mkDerivation rec {
+  pname = "libtensorflow";
+  version = "1.8.0";
+  name = "${pname}-${version}";
+  src = fetchurl {
+    url = "https://storage.googleapis.com/tensorflow/${pname}/${pname}-cpu-linux-x86_64-${version}.tar.gz";
+    sha256 = "0qzy15rc3x961cyi3bqnygrcnw4x69r28xkwhpwrv1r0gi6k73ha";
+  };
+
+  # Patch library to use our libc, libstdc++ and others
+  buildCommand = ''
+    . $stdenv/setup
+    mkdir -pv $out
+    tar -C $out -xzf $src
+    chmod +w $out/lib/libtensorflow.so
+    chmod +w $out/lib/libtensorflow_framework.so
+    ${patchelf}/bin/patchelf --set-rpath "${stdenv.cc.libc}/lib:${stdenv.cc.cc.lib}/lib:$out/lib" $out/lib/libtensorflow.so
+    ${patchelf}/bin/patchelf --set-rpath "${stdenv.cc.libc}/lib:${stdenv.cc.cc.lib}/lib" $out/lib/libtensorflow_framework.so
+    chmod -w $out/lib/libtensorflow.so
+    chmod -w $out/lib/libtensorflow_framework.so
+  '';
+
+  meta = with stdenv.lib; {
+    inherit version;
+    description = "C API for TensorFlow";
+    license = licenses.asl20;
+    maintainers = [maintainers.basvandijk];
+    platforms = platforms.linux;
+    homepage = https://www.tensorflow.org/versions/master/install/install_c;
+  };
+}
diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix
index bc4caf5988f1..3174dc66ca9d 100644
--- a/pkgs/top-level/all-packages.nix
+++ b/pkgs/top-level/all-packages.nix
@@ -860,6 +860,8 @@ with pkgs;
 
   tensor = libsForQt5.callPackage ../applications/networking/instant-messengers/tensor { };
 
+  libtensorflow = callPackage ../development/libraries/libtensorflow { };
+
   blink1-tool = callPackage ../tools/misc/blink1-tool { };
 
   bliss = callPackage ../applications/science/math/bliss { };