summarylogtreecommitdiffstats
path: root/PKGBUILD
blob: 8217ee4bc7ab2a6a8de8d51d15e556b3a7b432cf (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
37
38
39
# Maintainer: Micah Chambers <micahc.vt@gmail.com>

pkgname=caffe
pkgver=rc2
pkgrel=2
pkgdesc="Neural network toolkit"
arch=('i686' 'x86_64')
url="https://github.com/BVLC/caffe"
license=('BSD')
groups=('base-devel')
depends=('atlas-lapack-base' 'hdf5' 'opencv' 'boost' 'protobuf' 'google-glog'
		 'gflags' 'leveldb' 'snappy' 'lmdb')
makedepends=('git' 'cmake')
optdepends=('cuda: for GPU support')
source=("https://github.com/BVLC/$pkgname/archive/$pkgver.tar.gz")
sha256sums=('55c9c20870b30ce398e19e4f1a62ade1eff08fce51e28fa5604035b711978eec')

build() {
    if [ $CARCH = "i686" ]; then
        lib="lib"
    else
        lib="lib64"
    fi

    cd "$srcdir/$pkgname-$pkgver"
    rm -fr build
    mkdir build
    cd build
    cmake ../   -DAtlas_LAPACK_LIBRARY=/usr/${lib}/libatlas.so \
                -DAtlas_BLAS_LIBRARY=/usr/${lib}/libatlas.so \
                -DCMAKE_INSTALL_PREFIX=$pkgdir/usr/

    make
}

package() {
        cd "$srcdir/$pkgname-$pkgver/build"
        make install
}