summarylogtreecommitdiffstats
path: root/.SRCINFO
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