diff options
author | Leo Mao | 2018-12-13 17:06:11 +0800 |
---|---|---|
committer | Leo Mao | 2018-12-13 17:06:11 +0800 |
commit | 4dbc61c3d6bcbdf37c217e14aa81ed7fa2bfa3ca (patch) | |
tree | 0d8934e9e4d0585aa62c4ed892bc47aabc681e4a /PKGBUILD | |
parent | bd06ff4fcf6003c2d9a86f93f5765a12ccb59b3c (diff) | |
download | aur-python-torchvision-git.tar.gz |
upgpkg: python-torchvision-git 0.2.1.r62.g8bd05e6-1
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 60 |
1 files changed, 17 insertions, 43 deletions
@@ -1,16 +1,18 @@ -# Maintainer: Stephen Zhang <zsrkmyn at gmail dot com> - -pkgbase="python-torchvision-git" -pkgname=("python-torchvision-git" "python2-torchvision-git") -_pkgname="vision" -pkgver=0.1.8.r22.g85bb16a +# Maintainer: Leo Mao +pkgname=python-torchvision-git +_pkgname=vision +pkgver=0.2.1.r62.g8bd05e6 pkgrel=1 -pkgdesc="image and video datasets and models for torch deep learning" -arch=('x86_64') +pkgdesc="Datasets, Transforms and Models specific to Computer Vision" +arch=('any') url="https://github.com/pytorch/vision" +_github='pytorch/vision' license=('BSD') -makedepends=('python' 'python-setuptools' 'python2' 'python2-setuptools') -source=("git://github.com/pytorch/vision") +depends=('python' 'python-numpy' 'python-pillow' 'python-six' 'python-pytorch' 'python-tqdm') +makedepends=('python-setuptools') +provides=('python-torchvision') +conflicts=('python-torchvision') +source=("git+$url") sha256sums=('SKIP') pkgver () { @@ -21,44 +23,16 @@ pkgver () { ) } - -prepare() { - cd "$srcdir/" - - cp -a "${_pkgname}" "${_pkgname}-py2" - cd "${_pkgname}" - sed -e "s|#![ ]*/usr/bin/python$|#!/usr/bin/python2|" \ - -e "s|#![ ]*/usr/bin/env python$|#!/usr/bin/env python2|" \ - -e "s|#![ ]*/bin/env python$|#!/usr/bin/env python2|" \ - -i $(find . -name '*.py') -} - build() { - msg "Building Python 2" - cd "$srcdir/${_pkgname}-py2" - python2 setup.py build - msg "Building Python 3" - cd "$srcdir/${_pkgname}" + cd "${srcdir}/${_pkgname}" python setup.py build } -package_python2-torchvision-git() { - conflicts=('python2-torchvision') - provides=('python2-torchvision') - depends=('python2-pytorch') - cd "$srcdir/${_pkgname}-py2" - python2 setup.py install --root="$pkgdir"/ --optimize=1 --skip-build - install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE.txt" -} - -package_python-torchvision-git() { - conflicts=('python-torchvision') - provides=('python-torchvision') - depends=('python-pytorch') - cd "$srcdir/${_pkgname}" - python setup.py install --root="$pkgdir"/ --optimize=1 --skip-build - install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE.txt" +package() { + cd "${srcdir}/${_pkgname}" + python setup.py install --root="${pkgdir}"/ --optimize=1 --skip-build + install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE" } # vim:set ts=2 sw=2 et: |