20 lines
534 B
Nix
20 lines
534 B
Nix
pkgs: self: super: with self; {
|
|
|
|
# Backport upstream master tensorflow (bazel build is failing)
|
|
astunparse = callPackage ./tf-astunparse.nix { };
|
|
|
|
gast = callPackage ./tf-gast.nix { };
|
|
|
|
tensorflow-tensorboard = callPackage ./tf-tensorboard.nix {
|
|
protobuf = protobuf3_5;
|
|
};
|
|
|
|
tensorflow = callPackage ./tf.nix {
|
|
protobuf = protobuf3_5;
|
|
cudaSupport = pkgs.config.cudaSupport or false;
|
|
cudatoolkit = pkgs.cudatoolkit90;
|
|
cudnn = pkgs.cudnn_cudatoolkit90;
|
|
inherit (pkgs.linuxPackages) nvidia_x11;
|
|
};
|
|
}
|