zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Py keras extension (#3468)

Browse Source 
* Added support for building Theano with GPU support.  Added a package
for the libgpuarry libary needed for Theano.

* Added dependency on py-pexpect package which ipython requires.
This commit is contained in:
Brian Van Essen
2017-03-16 21:24:36 -07:00
committed by becker33
parent 0b27a7e13d
commit 3f17f761fe
3 changed files with 51 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
var/spack/repos/builtin/packages/py-theano/package.py
View File

@@ -34,12 +34,20 @@ class PyTheano(PythonPackage):
version('0.8.2', 'f2d0dfe7df141115201077cd933b2c52')
variant('gpu', default=False,
description='Builds with support for GPUs via CUDA and cuDNN')
depends_on('python@2.6:2.8,3.3:')
depends_on('py-setuptools', type='build')
depends_on('py-scipy@0.11:', type=('build', 'run'))
depends_on('py-numpy@1.7.1:', type=('build', 'run'))
depends_on('py-six@1.9.0:', type=('build', 'run'))
depends_on('blas')
depends_on('cuda', when='+gpu')
depends_on('libgpuarray', when='+gpu')
# test requirements
# depends_on('py-nose@1.3.0:', type=('build', 'run'))
# depends_on('py-nose-parameterized@0.5.0:', type=('build', 'run'))