about summary refs log tree commit diff
path: root/pkgs/development/python-modules/theano/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/theano/default.nix')
-rw-r--r--pkgs/development/python-modules/theano/default.nix5
1 files changed, 3 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/theano/default.nix b/pkgs/development/python-modules/theano/default.nix
index 1bcbda650a252..3a1af4a3517b3 100644
--- a/pkgs/development/python-modules/theano/default.nix
+++ b/pkgs/development/python-modules/theano/default.nix
@@ -11,8 +11,9 @@
 , setuptools
 , six
 , libgpuarray
-, cudaSupport ? false, cudaPackages ? {}
-, cudnnSupport ? false
+, config
+, cudaSupport ? config.cudaSupport or false, cudaPackages ? {}
+, cudnnSupport ? cudaSupport
 }:
 
 let