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:
Frederik Rietdijk 2017-06-02 09:07:19 +02:00 committed by GitHub
commit 474be6ff27
1 changed files with 2 additions and 1 deletions

View File

@ -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;
} }