1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-16 07:59:46 +01:00
tensorflow-haskell/shell.nix
2020-02-15 21:45:40 -08:00

19 lines
589 B
Nix

{ghc ? null }:
let
# Use pinned packages
nixpkgs = with (builtins.fromJSON (builtins.readFile ./nix/src.json));
builtins.fetchTarball {
url = "https://github.com/${owner}/${repo}/archive/${rev}.tar.gz";
inherit sha256;
};
pkgs = import nixpkgs {};
in
pkgs.haskell.lib.buildStackProject {
# Either use specified GHC or use GHC 8.6.5 (which we need for LTS 14.4)
ghc = if isNull ghc then pkgs.haskell.compiler.ghc865 else ghc;
extraArgs = "--system-ghc";
name = "tf-env";
buildInputs = with pkgs; [ snappy zlib protobuf libtensorflow ];
}