Syzygial/ccpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Theano: Backport master as this one can't find libnvrtc.so

Browse Source
This commit is contained in:
Tyson Whitehead
2018-08-10 17:44:42 -04:00
parent b6f66a65f5
commit 4868e74714
3 changed files with 196 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
temporary/python-packages.nix
View File

@@ -16,4 +16,29 @@ pkgs: self: super: with self; {
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;
};
}