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

py-horovod: add py-tensorflow-estimator dependency (#28674)

Browse Source
This commit is contained in:
Adam J. Stewart 2022-01-29 20:22:19 -06:00 committed by GitHub
parent c78a754e5d
commit 1679274ac2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
var/spack/repos/builtin/packages/py-horovod/package.py
View File

@ -63,6 +63,7 @@ class PyHorovod(PythonPackage, CudaPackage):
# Framework dependencies
depends_on('py-tensorflow@1.1.0:', type=('build', 'link', 'run'), when='frameworks=tensorflow')
depends_on('py-tensorflow@1.15:', type=('build', 'link', 'run'), when='frameworks=tensorflow @0.20:')
depends_on('py-tensorflow-estimator', type=('build', 'run'), when='frameworks=tensorflow')
depends_on('py-torch@0.4.0:', type=('build', 'link', 'run'), when='frameworks=pytorch')
depends_on('py-torch@1.2:', type=('build', 'link', 'run'), when='frameworks=pytorch @0.20:')
depends_on('py-torchvision', type=('build', 'run'), when='frameworks=pytorch @:0.19.1')