summarylogtreecommitdiffstats
diff options
context:
space:
mode:
authorJingbei Li2017-10-03 03:16:12 +0800
committerJingbei Li2017-10-03 03:16:12 +0800
commitd8234fa5897c36e4025e97ed760090c930990071 (patch)
tree7bbf89bcf9a00a5c73e000a2de06e276aff4e678
downloadaur-d8234fa5897c36e4025e97ed760090c930990071.tar.gz
Initial commit
-rw-r--r--.SRCINFO17
-rw-r--r--.gitignore5
-rw-r--r--PKGBUILD51
3 files changed, 73 insertions, 0 deletions
diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..bc4590e5d230
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,17 @@
+# Generated by mksrcinfo v8
+# Mon Oct 2 19:16:08 UTC 2017
+pkgbase = cntk
+ pkgdesc = Microsoft Cognitive Toolkit (CNTK), an open source deep-learning toolkit
+ pkgver = 2.2
+ pkgrel = 1
+ url = https://github.com/Microsoft/CNTK
+ arch = x86_64
+ license = CUSTOM
+ makedepends = python
+ makedepends = openblas
+ makedepends = swig
+ source = git+https://github.com/Microsoft/CNTK#tag=v2.2
+ md5sums = SKIP
+
+pkgname = cntk
+
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 000000000000..2269feac6eec
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,5 @@
+*/
+*.sw*
+*.pkg*
+*.tar.gz
+*.log
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..ae18d080f01c
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,51 @@
+# Maintainer: Jingbei Li <i@jingbei.li>
+pkgname=cntk
+_gitname=CNTK
+pkgver=2.2
+pkgrel=1
+pkgdesc="Microsoft Cognitive Toolkit (CNTK), an open source deep-learning toolkit"
+arch=('x86_64')
+url="https://github.com/Microsoft/$_gitname"
+license=('CUSTOM')
+depends=()
+optdepends=()
+makedepends=('python' 'openblas' 'swig')
+source=("git+$url#tag=v$pkgver")
+md5sums=('SKIP')
+
+prepare(){
+ cd $srcdir/$_gitname
+ git submodule update --init --recursive
+ sed '24a#include <cmath>' -i Source/CNTKv2LibraryDll/API/CNTKLibrary.h
+ sed \
+ -e 's|libprotobuf.a|libprotobuf.so|' \
+ -i configure
+ # -e 's|cuda/include/cudnn.h|include/cudnn.h|' \
+ mkdir build
+ ./configure \
+ --with-build-top=build \
+ --python=yes \
+ --with-jdk=/usr/lib/jvm/default \
+ --with-openblas \
+ --with-opencv
+ # --with-cuda=/opt/cuda \
+ # --with-cub=/usr/include \
+ # --with-gdk-include=/opt/cuda/include \
+ # --with-gdk-nvml-lib=/opt/cuda/lib64/stubs/ \
+ # --with-cudnn=/opt/cudnn6 \
+ # --with-nccl=/opt/cuda \
+ sed \
+ -e 's| $(PROTOBUF_PATH)/lib/libprotobuf.a| -lprotobuf|' \
+ -i Makefile
+}
+
+build() {
+ cd $srcdir/$_gitname/build
+ make CXXFLAGS='-Wno-sign-compare -fPIC'
+}
+
+package() {
+ mkdir -p $pkgdir/usr
+ cd $srcdir/$_gitname/build
+ cp -r bin lib $pkgdir/usr
+}