blob: 0933f14545e1ce1be40d826f0f86980e486c6544 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
# Maintainer: Carl Åkerlindh <carl.akerlindh at gmail dot com>
pkgname=mxnet
pkgver=0.9.3
pkgrel=1
pkgdesc="Flexible and Efficient Library for Deep Learning"
arch=('x86_64')
url="http://mxnet.io/"
license=('Apache2')
depends=(cuda cudnn)
# makedepends=()
# checkdepends=()
# optdepends=()
# provides=()
source=("$pkgname::git+https://github.com/dmlc/mxnet")
md5sums=('SKIP')
prepare() {
cd "$pkgname"
git checkout "v$pkgver"
git submodule update --init --recursive
cp make/config.mk .
echo "USE_CUDA=1" >>config.mk
echo "USE_CUDA_PATH=/opt/cuda" >>config.mk
echo "USE_CUDNN=1" >>config.mk
}
build() {
cd "$srcdir/$pkgname"
make -j$(nproc)
}
package() {
cd "$pkgdir"
install -D "$srcdir/$pkgname/lib/libmxnet.so" "$pkgdir/usr/lib/libmxnet.so"
}
|