mirror of
https://github.com/tensorflow/haskell.git
synced 2024-11-18 17:09:43 +01:00
7720af0afd
* Tested on linux without Docker. * Couldn't get nix build to work, so I just updated the URL and hash. * Did not test macos build. The mnist change was necessary because the argmax output type is now polmorphic.
65 lines
1.5 KiB
Nix
65 lines
1.5 KiB
Nix
#
|
|
# This script is provided by @grwlf (grwlf@gmail.com) and is supported on best
|
|
# effort basis. TensorFlow development team does not regularly test this script
|
|
# and can't answer any questions about it.
|
|
#
|
|
{ pkgs ? import <nixpkgs> {} }:
|
|
|
|
with pkgs;
|
|
|
|
let
|
|
tensorflow-c = stdenv.mkDerivation {
|
|
|
|
name = "tensorflow-c";
|
|
|
|
src = fetchurl {
|
|
url = "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.3.0.tar.gz";
|
|
sha256 = "1d4bda5316063b70cf50a668d774b2067ef2a8ab163ff2eb29592bf3c24e2183";
|
|
};
|
|
|
|
buildCommand = ''
|
|
. $stdenv/setup
|
|
mkdir -pv $out
|
|
tar -C $out -xzf $src
|
|
'';
|
|
};
|
|
|
|
fhs = pkgs.buildFHSUserEnv {
|
|
name = "fhs";
|
|
targetPkgs = pkgs:
|
|
with pkgs.python3Packages;
|
|
with pkgs.haskellPackages;
|
|
with pkgs;
|
|
assert stack.version >= "1.4";
|
|
assert ghc.version >= "8";
|
|
[
|
|
snappy
|
|
snappy.dev
|
|
protobuf3_2
|
|
stack
|
|
ghc
|
|
iana-etc
|
|
pkgs.zlib
|
|
pkgs.zlib.dev
|
|
tensorflow-c
|
|
gcc
|
|
gdb
|
|
];
|
|
runScript = "bash";
|
|
profile = ''
|
|
export USE_CCACHE=1
|
|
export LIBRARY_PATH=/usr/lib64
|
|
export LD_LIBRARY_PATH=$(pwd):/usr/lib64
|
|
export ACLOCAL_PATH=/usr/share/aclocal
|
|
export LANG=C
|
|
export LC_ALL=en_US.utf-8
|
|
export PATH="$HOME/.local/bin:$PATH"
|
|
'';
|
|
};
|
|
|
|
in
|
|
stdenv.mkDerivation {
|
|
name = fhs.name + "-env";
|
|
nativeBuildInputs = [ fhs ];
|
|
shellHook = "exec fhs";
|
|
}
|