diff options
author | lilac | 2020-12-02 21:14:21 +0800 |
---|---|---|
committer | lilac | 2020-12-02 21:14:21 +0800 |
commit | 8a807099298454db84c54663ee8b6760a06e250f (patch) | |
tree | 2400771ed1f47ac7b6bab025431a26570f80afa9 | |
parent | f7d4184768bbaa9315c0268f7aeec6ab9ff010f7 (diff) | |
download | aur-8a807099298454db84c54663ee8b6760a06e250f.tar.gz |
[lilac] updated to 0.8.1-3
-rw-r--r-- | .SRCINFO | 2 | ||||
-rw-r--r-- | PKGBUILD | 7 |
2 files changed, 5 insertions, 4 deletions
@@ -1,7 +1,7 @@ pkgbase = python-torchvision pkgdesc = Datasets, transforms, and models specific to computer vision pkgver = 0.8.1 - pkgrel = 2 + pkgrel = 3 url = https://github.com/pytorch/vision arch = x86_64 license = BSD @@ -4,10 +4,11 @@ # Based on python-torchvision-git; original contributors: # Contributor: Stephen Zhang <zsrkmyn at gmail dot 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=('python-torchvision' 'python-torchvision-cuda') _pkgname=vision pkgver=0.8.1 -pkgrel=2 +pkgrel=3 pkgdesc='Datasets, transforms, and models specific to computer vision' arch=('x86_64') url='https://github.com/pytorch/vision' @@ -54,7 +55,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" \ + TORCH_CUDA_ARCH_LIST=${_CUDA_ARCH_LIST} \ python setup.py build } @@ -82,7 +83,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" \ + TORCH_CUDA_ARCH_LIST=${_CUDA_ARCH_LIST} \ python setup.py install --root="${pkgdir}" --optimize=1 --skip-build install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}" } |