diff options
Diffstat (limited to 'pkgs/development/libraries/libtensorflow/default.nix')
-rw-r--r-- | pkgs/development/libraries/libtensorflow/default.nix | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/pkgs/development/libraries/libtensorflow/default.nix b/pkgs/development/libraries/libtensorflow/default.nix new file mode 100644 index 0000000000000..fce15b121b426 --- /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; + }; +} |