1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-01 00:29:42 +01:00
tensorflow-haskell/shell.nix

20 lines
590 B
Nix
Raw Normal View History

{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 {
2019-04-12 04:27:15 +02:00
# Either use specified GHC or use GHC 8.6.4 (which we need for LTS 13.13)
ghc = if isNull ghc then pkgs.haskell.compiler.ghc864 else ghc;
extraArgs = "--system-ghc";
name = "tf-env";
buildInputs = with pkgs; [ snappy zlib protobuf libtensorflow ];
}