summarylogtreecommitdiffstats
path: root/PKGBUILD
diff options
context:
space:
mode:
Diffstat (limited to 'PKGBUILD')
-rw-r--r--PKGBUILD6
1 files changed, 4 insertions, 2 deletions
diff --git a/PKGBUILD b/PKGBUILD
index ca75466fb4b..c2c17fcf927 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')