diff options
author | lilac | 2023-05-13 04:48:59 +0800 |
---|---|---|
committer | lilac | 2023-05-13 04:48:59 +0800 |
commit | 15e4eddcaecba90ab9160a89cdd6ff6a256d131b (patch) | |
tree | 6455736e90e3ccd9c147218f126371b558b03c15 | |
parent | b2323568bdfc2194c22202063167213ca737166d (diff) | |
download | aur-15e4eddcaecba90ab9160a89cdd6ff6a256d131b.tar.gz |
[lilac] updated to 0.28.0-1
-rw-r--r-- | .SRCINFO | 8 | ||||
-rw-r--r-- | PKGBUILD | 23 |
2 files changed, 19 insertions, 12 deletions
@@ -1,16 +1,18 @@ pkgbase = python-horovod pkgdesc = Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet - pkgver = 0.27.0 + pkgver = 0.28.0 pkgrel = 1 url = https://github.com/horovod/horovod arch = x86_64 license = Apache makedepends = cmake makedepends = git - makedepends = python-pip + makedepends = python-build + makedepends = python-installer makedepends = python-pytorch-cuda makedepends = python-setuptools makedepends = python-tensorflow-cuda + makedepends = python-wheel depends = cuda depends = cudnn depends = nccl @@ -27,7 +29,7 @@ pkgbase = python-horovod depends = python-tqdm optdepends = python-pytorch-cuda: pytorch framework optdepends = python-tensorflow-cuda: tensorflow framework - source = horovod-0.27.0::git+https://github.com/horovod/horovod.git#tag=v0.27.0 + source = horovod-0.28.0::git+https://github.com/horovod/horovod.git#tag=v0.28.0 sha512sums = SKIP pkgname = python-horovod @@ -2,7 +2,7 @@ _pkgname=horovod pkgname=python-horovod -pkgver=0.27.0 +pkgver=0.28.0 pkgrel=1 pkgdesc='Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet' arch=('x86_64') @@ -27,10 +27,12 @@ depends=( makedepends=( cmake git - python-pip + python-build + python-installer python-pytorch-cuda python-setuptools python-tensorflow-cuda + python-wheel ) optdepends=( 'python-pytorch-cuda: pytorch framework' @@ -43,29 +45,32 @@ prepare() { cd "${_pkgname}-${pkgver}" git submodule update --init --recursive # modify these environment variable as you need, see also https://github.com/horovod/horovod/blob/master/docs/install.rst - export HOROVOD_CUDA_HOME=/opt/cuda + export HOROVOD_BUILD_CUDA_CC_LIST="60,61,62,70,72,75,80,86,89,90" export HOROVOD_CPU_OPERATIONS=GLOO + export HOROVOD_CUDA_HOME=/opt/cuda export HOROVOD_GPU=CUDA - export HOROVOD_GPU_ALLREDUCE=NCCL - export HOROVOD_GPU_BROADCAST=NCCL + export HOROVOD_GPU_OPERATIONS=NCCL + export HOROVOD_NCCL_LINK=SHARED + export HOROVOD_WITHOUT_MXNET=1 export HOROVOD_WITH_GLOO=1 export HOROVOD_WITH_MPI=1 - export HOROVOD_WITH_MXNET=0 export HOROVOD_WITH_PYTORCH=1 export HOROVOD_WITH_TENSORFLOW=1 - export HOROVOD_BUILD_CUDA_CC_LIST="52,53,60,61,62,70,72,75,80,86" export CC=/opt/cuda/bin/gcc export CXX=/opt/cuda/bin/g++ + # fix https://github.com/horovod/horovod/issues/3923 + cd "third_party/gloo" + git pull https://github.com/facebookincubator/gloo.git } build() { cd "${_pkgname}-${pkgver}" - python setup.py build + python -m build --wheel --no-isolation -x } package() { cd "${_pkgname}-${pkgver}" - python setup.py install --root="${pkgdir}" --optimize=1 --skip-build + python -m installer --destdir="${pkgdir}" dist/*.whl install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}" } # vim:set ts=2 sw=2 et: |