summarylogtreecommitdiffstats
diff options
context:
space:
mode:
authorlilac2020-10-05 19:39:19 +0800
committerlilac2020-10-05 19:39:19 +0800
commit041034f6e7c023982712950825c335b93cb37291 (patch)
tree7b2ddba7ce6fc7761ef9fb421511e36809c0261a
parent8d9e4bb9588750cdce971a0e15697c42069f1d8e (diff)
downloadaur-041034f6e7c023982712950825c335b93cb37291.tar.gz
[lilac] updated to 0.7.0-4
-rw-r--r--.SRCINFO3
-rw-r--r--PKGBUILD7
2 files changed, 6 insertions, 4 deletions
diff --git a/.SRCINFO b/.SRCINFO
index 75912f0d15c..05ba9b8be14 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,7 +1,7 @@
pkgbase = python-torchvision
pkgdesc = Datasets, transforms, and models specific to computer vision
pkgver = 0.7.0
- pkgrel = 2
+ pkgrel = 4
url = https://github.com/pytorch/vision
arch = x86_64
license = BSD
@@ -21,6 +21,7 @@ pkgbase = python-torchvision
depends = python-tqdm
optdepends = ffmpeg: video reader backend (the recommended one with better performance)
optdepends = python-av: video reader backend (the default one)
+ optdepends = python-pycocotools: support for MS-COCO dataset
source = vision-0.7.0.tar.gz::https://github.com/pytorch/vision/archive/v0.7.0.tar.gz
sha512sums = f22c8fd67f4efafa4a934ed5014e1ad661d0d0a800d4c8ae0e6b85f4860a5dc131c513428028949c9130486ba24ecf8a9f6b9881efc68a0c2158110924d05e81
diff --git a/PKGBUILD b/PKGBUILD
index b3ee63cf359..2ee1e5549f3 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -7,7 +7,7 @@
pkgname=('python-torchvision' 'python-torchvision-cuda')
_pkgname=vision
pkgver=0.7.0
-pkgrel=2
+pkgrel=4
pkgdesc='Datasets, transforms, and models specific to computer vision'
arch=('x86_64')
url='https://github.com/pytorch/vision'
@@ -22,6 +22,7 @@ depends=(
optdepends=(
'ffmpeg: video reader backend (the recommended one with better performance)'
'python-av: video reader backend (the default one)'
+ 'python-pycocotools: support for MS-COCO dataset'
)
makedepends=(
'cuda'
@@ -53,7 +54,7 @@ build() {
cd "${srcdir}/${_pkgname}-cuda-${pkgver}"
FORCE_CUDA=1 \
- TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0;8.0+PTX" \
+ TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0" \
python setup.py build
}
@@ -81,7 +82,7 @@ package_python-torchvision-cuda() {
cd "${srcdir}/${_pkgname}-cuda-${pkgver}"
FORCE_CUDA=1 \
- TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0;8.0+PTX" \
+ TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0" \
python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}"
}