diff options
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 6 |
1 files changed, 4 insertions, 2 deletions
@@ -4,12 +4,14 @@ pkgbase="python-pytorch-git" pkgname=("python-pytorch-git" "python2-pytorch-git") _pkgname="pytorch" pkgver=0.1.10.r46.gbe6322e4 -pkgrel=1 +pkgrel=2 pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration" arch=('x86_64') url="https://github.com/pytorch/pytorch" license=('BSD') -makedepends=('python' 'python-setuptools' 'python2' 'python2-setuptools' 'gcc5' 'cmake' 'git') +makedepends=('python' 'python-setuptools' 'python2' 'python2-setuptools' + 'python-yaml' 'python2-yaml' 'python-numpy' 'python2-numpy' + 'gcc5' 'cmake' 'git') depends=('cuda' 'cudnn') source=("git://github.com/pytorch/pytorch") sha256sums=('SKIP') |