3
0
Fork 0
forked from mirrors/nixpkgs

pythonPackages.Theano: fix cuda package

* add missing setuptools for cuda
* remove unused nvidia_x11
This commit is contained in:
Ryan Burns 2021-09-21 01:07:45 -07:00
parent 11bf0ac25b
commit 63391899a3
2 changed files with 10 additions and 7 deletions

View file

@ -8,19 +8,15 @@
, nose
, numpy
, scipy
, setuptools
, six
, libgpuarray
, cudaSupport ? false, cudatoolkit
, cudnnSupport ? false, cudnn
, nvidia_x11
}:
assert cudnnSupport -> cudaSupport;
assert cudaSupport -> nvidia_x11 != null
&& cudatoolkit != null
&& cudnn != null;
let
wrapped = command: buildTop: buildInputs:
runCommandCC "${command}-wrapped" { inherit buildInputs; } ''
@ -81,7 +77,15 @@ in buildPythonPackage rec {
# keep Nose around since running the tests by hand is possible from Python or bash
checkInputs = [ nose ];
propagatedBuildInputs = [ numpy numpy.blas scipy six libgpuarray_ ];
# setuptools needed for cuda support
propagatedBuildInputs = [
libgpuarray_
numpy
numpy.blas
scipy
setuptools
six
];
pythonImportsCheck = [ "theano" ];

View file

@ -8881,7 +8881,6 @@ in {
Theano = callPackage ../development/python-modules/Theano rec {
cudaSupport = pkgs.config.cudaSupport or false;
cudnnSupport = cudaSupport;
inherit (pkgs.linuxPackages) nvidia_x11;
};
TheanoWithCuda = self.Theano.override {