py-horovod: link dependency for py-torch (#16600)

fix for same reasons as #16495.
This commit is contained in:
Andrew W Elble 2020-05-12 12:24:30 -04:00 committed by GitHub
parent 7813e41464
commit e36aa96392
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -46,7 +46,7 @@ class PyHorovod(PythonPackage):
# Framework dependencies
depends_on('py-tensorflow@1.1.0:', type=('build', 'link', 'run'), when='frameworks=tensorflow')
depends_on('py-torch@0.4.0:', type=('build', 'run'), when='frameworks=pytorch')
depends_on('py-torch@0.4.0:', type=('build', 'link', 'run'), when='frameworks=pytorch')
depends_on('py-torchvision', type=('build', 'run'), when='frameworks=pytorch')
depends_on('py-cffi@1.4.0:', type=('build', 'run'), when='frameworks=pytorch')
depends_on('mxnet@1.4.1:+python', type=('build', 'link', 'run'), when='frameworks=mxnet')