diff options
author | lilac | 2020-08-04 04:53:20 +0800 |
---|---|---|
committer | lilac | 2020-08-04 04:53:20 +0800 |
commit | 8d9e4bb9588750cdce971a0e15697c42069f1d8e (patch) | |
tree | 617d3d2055e292428276002f320484386cb5a491 /PKGBUILD | |
parent | 6493d9226bda5bee4aba29bd5b65ca2e7ac8134f (diff) | |
download | aur-8d9e4bb9588750cdce971a0e15697c42069f1d8e.tar.gz |
[lilac] updated to 0.7.0-2
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 10 |
1 files changed, 5 insertions, 5 deletions
@@ -6,8 +6,8 @@ pkgname=('python-torchvision' 'python-torchvision-cuda') _pkgname=vision -pkgver=0.6.1 -pkgrel=1 +pkgver=0.7.0 +pkgrel=2 pkgdesc='Datasets, transforms, and models specific to computer vision' arch=('x86_64') url='https://github.com/pytorch/vision' @@ -37,7 +37,7 @@ checkdepends=( 'python-scipy' ) source=("${_pkgname}-${pkgver}.tar.gz::https://github.com/pytorch/vision/archive/v${pkgver}.tar.gz") -sha512sums=('428235ae9412d176dc62eb42ed926caa5199bab448e7798fa240d69c1337e1ca5423c45a34bd6f4527e7ccafb927f53743311f8701dcee5f168bacc60f1d6750') +sha512sums=('f22c8fd67f4efafa4a934ed5014e1ad661d0d0a800d4c8ae0e6b85f4860a5dc131c513428028949c9130486ba24ecf8a9f6b9881efc68a0c2158110924d05e81') get_pyver() { python -c 'import sys; print(str(sys.version_info[0]) + "." + str(sys.version_info[1]))' @@ -53,7 +53,7 @@ build() { cd "${srcdir}/${_pkgname}-cuda-${pkgver}" FORCE_CUDA=1 \ - TORCH_CUDA_ARCH_LIST="3.5;3.7;5.0;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" \ + 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" \ python setup.py build } @@ -81,7 +81,7 @@ package_python-torchvision-cuda() { cd "${srcdir}/${_pkgname}-cuda-${pkgver}" FORCE_CUDA=1 \ - TORCH_CUDA_ARCH_LIST="3.5;3.7;5.0;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" \ + 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" \ python setup.py install --root="${pkgdir}" --optimize=1 --skip-build install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}" } |