blob: 1ed881cb3fe2defb12d354805340ece0a9c1944c (
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
# Maintainer: Butui Hu <hot123tea123@gmail.com>
_CUDA_ARCH_LIST="5.2;5.3;6.0;6.1;6.2;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0;8.0+PTX;8.6;8.6+PTX"
_pkgname=mmcv
pkgname=(python-mmcv python-mmcv-full)
pkgver=1.3.12
pkgrel=1
pkgdesc='OpenMMLab Computer Vision Foundation'
arch=('x86_64')
url='https://github.com/open-mmlab/mmcv'
license=('Apache')
depends=(
python-addict
python-lmdb
python-opencv
python-pillow
)
makedepends=(
cuda
cython
python-pip
python-pytorch-cuda
python-setuptools
)
source=("${pkgname}-${pkgver}.tar.gz::https://github.com/open-mmlab/mmcv/archive/v${pkgver}.tar.gz")
sha512sums=('7101ffbcf4921fb3aa01662dfb8f483e2d08afbe9e63f81ed4e0b2fd7a5f82077fa2e3f837647f6dcc76983f6679c7a667eae245ed64435c4367e43af99db722')
prepare() {
cp -a "${_pkgname}-${pkgver}" "${_pkgname}-full-${pkgver}"
}
build() {
cd "${srcdir}/${_pkgname}-${pkgver}"
MMCV_WITH_OPS=1 python setup.py build_py
MMCV_WITH_OPS=1 python setup.py build_ext
cd "${srcdir}/${_pkgname}-full-${pkgver}"
FORCE_CUDA=1 \
MMCV_WITH_OPS=1 \
TORCH_CUDA_ARCH_LIST=${_CUDA_ARCH_LIST} \
python setup.py build_py
FORCE_CUDA=1 \
MMCV_WITH_OPS=1 \
TORCH_CUDA_ARCH_LIST=${_CUDA_ARCH_LIST} \
python setup.py build_ext
}
package_python-mmcv() {
pkgdesc+="(cpu version, without cuda ops)"
depends+=(python-pytorch)
cd "${_pkgname}-${pkgver}"
MMCV_WITH_OPS=1 python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
}
package_python-mmcv-full() {
pkgdesc+=" (full version, with full features, include cuda ops)"
depends+=(cuda python-pytorch-cuda)
provides=(python-mmcv=${pkgver})
conflicts=(python-mmcv)
cd "${_pkgname}-full-${pkgver}"
FORCE_CUDA=1 \
MMCV_WITH_OPS=1 \
TORCH_CUDA_ARCH_LIST=${_CUDA_ARCH_LIST} \
python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
}
# vim:set ts=2 sw=2 et:
|