diff options
-rw-r--r-- | .SRCINFO | 13 | ||||
-rw-r--r-- | PKGBUILD | 8 |
2 files changed, 14 insertions, 7 deletions
@@ -1,7 +1,7 @@ pkgbase = python-torchvision pkgdesc = Datasets, transforms, and models specific to computer vision pkgver = 0.5.0 - pkgrel = 2 + pkgrel = 4 url = https://github.com/pytorch/vision arch = x86_64 license = BSD @@ -11,11 +11,11 @@ pkgbase = python-torchvision makedepends = cuda makedepends = ffmpeg makedepends = python-av + makedepends = python-pytorch-cuda makedepends = python-setuptools makedepends = qt5-base depends = python-numpy depends = python-pillow - depends = python-pytorch-cuda depends = python-scipy depends = python-six depends = python-tqdm @@ -25,16 +25,21 @@ pkgbase = python-torchvision sha512sums = 2b76f7a0f20b7225f6310be358bf4c6fb8cdbc1fa324f53dac3bf6518276755861bd09a91da6cf6a0ff1b466104826b4094e82c796c83b9b4963ca3c0b94d93a pkgname = python-torchvision + depends = python-numpy + depends = python-pillow + depends = python-scipy + depends = python-six + depends = python-tqdm + depends = python-pytorch pkgname = python-torchvision-cuda pkgdesc = Datasets, transforms, and models specific to computer vision (with GPU support) depends = python-numpy depends = python-pillow - depends = python-pytorch-cuda depends = python-scipy depends = python-six depends = python-tqdm - depends = cuda + depends = python-pytorch-cuda provides = python-torchvision=0.5.0 conflicts = python-torchvision=0.5.0 @@ -7,7 +7,7 @@ pkgname=('python-torchvision' 'python-torchvision-cuda') _pkgname=vision pkgver=0.5.0 -pkgrel=2 +pkgrel=4 pkgdesc='Datasets, transforms, and models specific to computer vision' arch=('x86_64') url='https://github.com/pytorch/vision' @@ -15,7 +15,6 @@ license=('BSD') depends=( 'python-numpy' 'python-pillow' - 'python-pytorch-cuda' 'python-scipy' 'python-six' 'python-tqdm' @@ -28,6 +27,7 @@ makedepends=( 'cuda' 'ffmpeg' 'python-av' + 'python-pytorch-cuda' 'python-setuptools' 'qt5-base' ) @@ -66,6 +66,8 @@ check() { } package_python-torchvision() { + depends+=(python-pytorch) + cd "${srcdir}/${_pkgname}-${pkgver}" python setup.py install --root="${pkgdir}" --optimize=1 --skip-build install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}" @@ -73,7 +75,7 @@ package_python-torchvision() { package_python-torchvision-cuda() { pkgdesc='Datasets, transforms, and models specific to computer vision (with GPU support)' - depends+=(cuda) + depends+=(python-pytorch-cuda) provides+=(python-torchvision=${pkgver}) conflicts+=(python-torchvision=${pkgver}) |