Merge pull request #26263 from twhitehead/theano-cuda-cudnn
TheanoWithCuda: cudnn isn't added to python environment due to use of optional with a list
This commit is contained in:
commit
474be6ff27
|
@ -56,7 +56,8 @@ buildPythonPackage rec {
|
||||||
pycuda
|
pycuda
|
||||||
cudatoolkit
|
cudatoolkit
|
||||||
libgpuarray
|
libgpuarray
|
||||||
] ++ (stdenv.lib.optional (cudnn != null) [ cudnn ]);
|
cudnn
|
||||||
|
];
|
||||||
|
|
||||||
passthru.cudaSupport = true;
|
passthru.cudaSupport = true;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue