blob: fe5175b8798171fcbcd451b284a9cd5b408c78ca (
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
36
37
38
39
40
41
42
43
44
45
46
47
|
# Maintainer: Butui Hu <hot123tea123@gmail.com>
pkgname=python-pytorch-lightning
_pkgname=pytorch-lightning
pkgver=1.7.2
pkgrel=1
pkgdesc="The lightweight PyTorch wrapper for high-performance AI research"
arch=('any')
url='https://lightning.ai'
license=('Apache')
depends=(
python-fsspec
python-jsonargparse
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=(
python-build
python-installer
python-wheel
)
source=("${_pkgname}-${pkgver}.tar.gz::https://files.pythonhosted.org/packages/source/${_pkgname::1}/${_pkgname}/${_pkgname}-${pkgver}.tar.gz")
sha512sums=('1a3c66c6b82551fcd12ce4fba500a623955a655e54f0009d818a178c8e2f2a2d8ec565f585624803da1c4978d486527e53e4c9b048eeb1c8110d87ec1ff3cc70')
build() {
cd "${_pkgname}-${pkgver}"
python -m build --wheel --no-isolation
}
package() {
cd "${_pkgname}-${pkgver}"
python -m installer --destdir="${pkgdir}" dist/*.whl
}
# vim:set ts=2 sw=2 et:
|