update pkg && fix makedepends

1 files changed, 4 insertions, 2 deletions

diff --git a/PKGBUILD b/PKGBUILD
index ca75466fb4b5..c2c17fcf9276 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -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')