diff options
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 7 |
1 files changed, 4 insertions, 3 deletions
@@ -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}" } |