pythonPackages.Theano: fix cuda package

* add missing setuptools for cuda
* remove unused nvidia_x11
This commit is contained in:
Ryan Burns
2021-09-22 23:29:17 -07:00
parent 11bf0ac25b
commit 63391899a3
2 changed files with 10 additions and 7 deletions
@@ -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" ];