diff options
-rw-r--r-- | var/spack/repos/builtin/packages/py-horovod/package.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-horovod/package.py b/var/spack/repos/builtin/packages/py-horovod/package.py index 6bd68530e5..eed7ca386d 100644 --- a/var/spack/repos/builtin/packages/py-horovod/package.py +++ b/var/spack/repos/builtin/packages/py-horovod/package.py @@ -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') |