diff options
-rw-r--r-- | .SRCINFO | 18 | ||||
-rw-r--r-- | PKGBUILD | 29 |
2 files changed, 47 insertions, 0 deletions
diff --git a/.SRCINFO b/.SRCINFO new file mode 100644 index 000000000000..2657552baa51 --- /dev/null +++ b/.SRCINFO @@ -0,0 +1,18 @@ +pkgbase = python-pytorch-inplace_abn + pkgdesc = In-Place Activated BatchNorm for Memory-Optimized Training of DNNs (Pytorch) + pkgver = 1.0.12 + pkgrel = 1 + url = https://github.com/mapillary/inplace_abn + arch = x86_64 + license = BSD + makedepends = python-setuptools + depends = python-pytorch + depends = python-pillow + depends = python-numpy + depends = python-tensorboardx + depends = python-torchvision + source = git+https://github.com/mapillary/inplace_abn.git#tag=v1.0.12 + sha256sums = SKIP + +pkgname = python-pytorch-inplace_abn + diff --git a/PKGBUILD b/PKGBUILD new file mode 100644 index 000000000000..39a6e28868e3 --- /dev/null +++ b/PKGBUILD @@ -0,0 +1,29 @@ +# Maintainer: trougnouf (Benoit Brummer) <trougnouf at gmail dot com> + +_name='inplace_abn' +pkgname="python-pytorch-${_name}" +pkgver=1.0.12 +pkgrel=1 +pkgdesc="In-Place Activated BatchNorm for Memory-Optimized Training of DNNs (Pytorch)" +url="https://github.com/mapillary/${_name}" +depends=('python-pytorch' 'python-pillow' 'python-numpy' 'python-tensorboardx' 'python-torchvision') +makedepends=('python-setuptools') +license=('BSD') +arch=('x86_64') + +source=("git+${url}.git#tag=v${pkgver}") + +sha256sums=('SKIP') + +build() { + cd "${srcdir}/${_name}" + python setup.py build +} + +package() { + cd "${srcdir}/${_name}" + python setup.py install --root="$pkgdir/" --optimize=1 --skip-build +} + + + |