blob: 4f3c42fbc9c90233cea4b18a91307e43302aedec (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
pkgbase = python-horovod
pkgdesc = Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet
pkgver = 0.21.0
pkgrel = 1
url = https://github.com/horovod/horovod
arch = x86_64
license = Apache
makedepends = cmake
makedepends = git
makedepends = mxnet-cuda
makedepends = python-pytorch-cuda
makedepends = python-setuptools
makedepends = python-tensorflow-cuda
depends = cuda
depends = cudnn
depends = nccl
depends = openmpi
depends = python-cloudpickle
depends = python-h5py
depends = python-mock
depends = python-mpi4py
depends = python-psutil
depends = python-pycparser
depends = python-pyaml
depends = python-scipy
depends = python-six
depends = python-tqdm
optdepends = mxnet-cuda: mxnet framework
optdepends = python-pytorch-cuda: pytorch framework
optdepends = python-tensorflow-cuda: tensorflow framework
source = horovod-0.21.0::git+https://github.com/horovod/horovod.git#tag=v0.21.0
sha512sums = SKIP
pkgname = python-horovod
|