temporary: Drop workarounds for previous upstream version
This commit is contained in:
@@ -1,44 +1,3 @@
|
||||
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;
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user