diff options
author | Frederik Rietdijk <fridh@fridh.nl> | 2017-02-15 12:30:30 +0100 |
---|---|---|
committer | Frederik Rietdijk <fridh@fridh.nl> | 2017-02-15 12:32:44 +0100 |
commit | a069c16c2240fecc69545b9b26c0becea8ac559d (patch) | |
tree | 8913566ef904aae2546a8d8b787f2556a9291abf /pkgs/development/python-modules/tensorflow | |
parent | f53900d142744344d31cc7fa40fc0e2db9c97ba6 (diff) | |
download | nixlib-a069c16c2240fecc69545b9b26c0becea8ac559d.tar nixlib-a069c16c2240fecc69545b9b26c0becea8ac559d.tar.gz nixlib-a069c16c2240fecc69545b9b26c0becea8ac559d.tar.bz2 nixlib-a069c16c2240fecc69545b9b26c0becea8ac559d.tar.lz nixlib-a069c16c2240fecc69545b9b26c0becea8ac559d.tar.xz nixlib-a069c16c2240fecc69545b9b26c0becea8ac559d.tar.zst nixlib-a069c16c2240fecc69545b9b26c0becea8ac559d.zip |
tensorflowNoGpuSupport -> tensorflowWithoutCuda
and move expression
Diffstat (limited to 'pkgs/development/python-modules/tensorflow')
-rw-r--r-- | pkgs/development/python-modules/tensorflow/default.nix | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/tensorflow/default.nix b/pkgs/development/python-modules/tensorflow/default.nix new file mode 100644 index 000000000000..f8bc35eb5684 --- /dev/null +++ b/pkgs/development/python-modules/tensorflow/default.nix @@ -0,0 +1,48 @@ +{ stdenv +, fetchurl +, buildPythonPackage +, numpy +, six +, protobuf3_0_0b2 +, swig +, mock +, gcc +, zlib +}: + +# tensorflow is built from a downloaded wheel, because the upstream +# project's build system is an arcane beast based on +# bazel. Untangling it and building the wheel from source is an open +# problem. + +buildPythonPackage rec { + pname = "tensorflow"; + version = "0.10.0"; + name = "${pname}-${version}"; + format = "wheel"; + + src = fetchurl { + url = if stdenv.isDarwin then + "https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-${version}-py2-none-any.whl" else + "https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-${version}-cp27-none-linux_x86_64.whl"; + sha256 = if stdenv.isDarwin then + "1gjybh3j3rn34bzhsxsfdbqgsr4jh50qyx2wqywvcb24fkvy40j9" else + "0g05pa4z6kdy0giz7hjgjgwf4zzr5l8cf1zh247ymixlikn3fnpx"; + }; + + propagatedBuildInputs = [ numpy six protobuf3_0_0b2 swig mock]; + + preFixup = '' + RPATH="${stdenv.lib.makeLibraryPath [ gcc.cc.lib zlib ]}" + find $out -name '*.so' -exec patchelf --set-rpath "$RPATH" {} \; + ''; + + doCheck = false; + + meta = with stdenv.lib; { + description = "TensorFlow helps the tensors flow (no gpu support)"; + homepage = http://tensorflow.org; + license = licenses.asl20; + platforms = with platforms; linux ++ darwin; + }; +} |