diff options
| -rw-r--r-- | .SRCINFO | 12 | ||||
| -rw-r--r-- | PKGBUILD | 10 |
2 files changed, 11 insertions, 11 deletions
@@ -1,7 +1,7 @@ pkgbase = python-torchvision pkgdesc = Datasets, transforms, and models specific to computer vision - pkgver = 0.6.1 - pkgrel = 1 + pkgver = 0.7.0 + pkgrel = 2 url = https://github.com/pytorch/vision arch = x86_64 license = BSD @@ -21,8 +21,8 @@ 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) - source = vision-0.6.1.tar.gz::https://github.com/pytorch/vision/archive/v0.6.1.tar.gz - sha512sums = 428235ae9412d176dc62eb42ed926caa5199bab448e7798fa240d69c1337e1ca5423c45a34bd6f4527e7ccafb927f53743311f8701dcee5f168bacc60f1d6750 + source = vision-0.7.0.tar.gz::https://github.com/pytorch/vision/archive/v0.7.0.tar.gz + sha512sums = f22c8fd67f4efafa4a934ed5014e1ad661d0d0a800d4c8ae0e6b85f4860a5dc131c513428028949c9130486ba24ecf8a9f6b9881efc68a0c2158110924d05e81 pkgname = python-torchvision depends = python-numpy @@ -40,6 +40,6 @@ pkgname = python-torchvision-cuda depends = python-six depends = python-tqdm depends = python-pytorch-cuda - provides = python-torchvision=0.6.1 - conflicts = python-torchvision=0.6.1 + provides = python-torchvision=0.7.0 + conflicts = python-torchvision=0.7.0 @@ -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}" } |