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