blob: d2992ba33ba2e4f57d79e5d0ec70e00d0d2bfa2f (
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
|
pkgbase = python-horovod
pkgdesc = Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet
pkgver = 0.19.1
pkgrel = 1
url = https://github.com/horovod/horovod
arch = x86_64
license = Apache
makedepends = cmake
makedepends = git
makedepends = mxnet-cuda
makedepends = python-keras-applications
makedepends = python-pytorch-cuda
makedepends = python-setuptools
makedepends = python-tensorflow-cuda
depends = cuda
depends = cudnn
depends = nccl
depends = openmpi
depends = python-h5py
depends = python-mock
depends = python-mpi4py
depends = python-scipy
depends = python-tqdm
optdepends = mxnet-cuda
optdepends = python-keras-applications
optdepends = python-pytorch-cuda
optdepends = python-tensorflow-cuda
source = horovod-0.19.1::git+https://github.com/horovod/horovod.git#tag=v0.19.1
sha512sums = SKIP
pkgname = python-horovod
|