diff options
| author | Philip Goto | 2021-12-14 13:06:53 +0100 |
|---|---|---|
| committer | Philip Goto | 2021-12-14 13:06:53 +0100 |
| commit | 9f143fdafe1fb01b5c706bc2180c408d320d4d0b (patch) | |
| tree | b0505becf46be72ff2a1a7ce4fe46fce83527c74 /PKGBUILD | |
| download | aur-9f143fdafe1fb01b5c706bc2180c408d320d4d0b.tar.gz | |
Initialize
Diffstat (limited to 'PKGBUILD')
| -rw-r--r-- | PKGBUILD | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/PKGBUILD b/PKGBUILD new file mode 100644 index 000000000000..65e945f475c2 --- /dev/null +++ b/PKGBUILD @@ -0,0 +1,51 @@ +# Maintainer: Philip Goto <philip.goto@gmail.com> +# Contributor: Butui Hu <hot123tea123@gmail.com> + +pkgname=python-pytorch-lightning-git +_pkgname=pytorch-lightning +pkgver=1.5.0.r244.g2faaf35b91 +pkgrel=1 +pkgdesc='The lightweight PyTorch wrapper for high-performance AI research' +arch=(any) +url='https://github.com/PyTorchLightning/pytorch-lightning' +license=('Apache') +depends=( + python-fsspec + python-pandas + python-pyaml + python-pydeprecate + python-numpy + python-scikit-learn + python-pytorch + python-torchmetrics + python-torchvision + python-twine + python-tqdm + tensorboard +) +optdepends=( + 'python-apex: mixed precision support' + 'python-horovod: for distributed training' +) +makedepends=(git python-setuptools) +provides=(python-pytorch-lightning) +conflicts=(python-pytorch-lightning) +source=("git+https://github.com/PyTorchLightning/pytorch-lightning.git") +sha512sums=(SKIP) + +pkgver() { + cd "${_pkgname}" + git describe --long --tags | sed 's/^v//;s/\([^-]*-g\)/r\1/;s/-/./g' +} + + +build() { + cd "${_pkgname}" + python setup.py build +} + +package() { + cd "${_pkgname}" + python setup.py install --skip-build --root="${pkgdir}" --optimize=1 + install -Dm644 LICENSE "$pkgdir/usr/share/licenses/${_pkgname}/LICENSE" +} |