45 lines
1.3 KiB
Nix
45 lines
1.3 KiB
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;
|
|
};
|
|
|
|
# https://github.com/NixOS/nixpkgs/pull/26580 (Cuda definitions are backwards too)
|
|
libgpuarray = callPackage ./theano-libgpuarray.nix {
|
|
clblas = pkgs.clblas.override { inherit boost; };
|
|
cudaSupport = pkgs.config.cudaSupport or false;
|
|
inherit (pkgs.linuxPackages) nvidia_x11;
|
|
};
|
|
|
|
Theano = callPackage ./theano.nix rec {
|
|
cudaSupport = pkgs.config.cudaSupport or false;
|
|
cudnnSupport = cudaSupport;
|
|
cudatoolkit = pkgs.cudatoolkit8;
|
|
cudnn = pkgs.cudnn_cudatoolkit8;
|
|
inherit (pkgs.linuxPackages) nvidia_x11;
|
|
};
|
|
|
|
TheanoWithoutCuda = super.TheanoWithoutCuda.override {
|
|
cudaSupport = false;
|
|
cudnnSupport = false;
|
|
};
|
|
|
|
TheanoWithCuda = super.TheanoWithCuda.override {
|
|
cudaSupport = true;
|
|
cudnnSupport = true;
|
|
};
|
|
}
|