diff options
author | Daniel Bermond | 2018-09-29 13:13:42 +0000 |
---|---|---|
committer | Daniel Bermond | 2018-09-29 13:13:42 +0000 |
commit | 361fdf017ae10a5991df2a6fe5f893b85ac85cdf (patch) | |
tree | 62b8db7f9192379a35fa2325735787eb971b1ca4 | |
parent | 213ddfea35bf7110130b1ecfb03420ef7d3b33c1 (diff) | |
download | aur-361fdf017ae10a5991df2a6fe5f893b85ac85cdf.tar.gz |
Add support for nvidia volta and turing architectures
-rw-r--r-- | .SRCINFO | 4 | ||||
-rw-r--r-- | Makefile.config | 4 | ||||
-rw-r--r-- | PKGBUILD | 4 |
3 files changed, 7 insertions, 5 deletions
@@ -1,7 +1,7 @@ pkgbase = caffe-cuda pkgdesc = A deep learning framework made with expression, speed, and modularity in mind (with cuda) pkgver = 1.0 - pkgrel = 2 + pkgrel = 3 url = http://caffe.berkeleyvision.org/ arch = x86_64 license = BSD @@ -52,7 +52,7 @@ pkgbase = caffe-cuda source = caffe-cuda-1.0.tar.gz::https://github.com/BVLC/caffe/archive/1.0.tar.gz source = Makefile.config sha256sums = 71d3c9eb8a183150f965a465824d01fe82826c22505f7aa314f700ace03fa77f - sha256sums = 9426828e7598bc85ae0ded53779643d99072e4e9c49f8a8b1deab8c166cbc2e7 + sha256sums = c61207ffea4ade927ce66a980f44732c1c2f9b82c500d51e687f5bcf773a4465 pkgname = caffe-cuda diff --git a/Makefile.config b/Makefile.config index 5fe8110f4cca..901d7c88b680 100644 --- a/Makefile.config +++ b/Makefile.config @@ -11,7 +11,9 @@ CUDA_ARCH := -gencode arch=compute_30,code=sm_30 \ -gencode arch=compute_52,code=sm_52 \ -gencode arch=compute_60,code=sm_60 \ -gencode arch=compute_61,code=sm_61 \ - -gencode arch=compute_61,code=compute_61 + -gencode arch=compute_70,code=sm_70 \ + -gencode arch=compute_75,code=sm_75 \ + -gencode arch=compute_75,code=compute_75 BLAS := open BLAS_INCLUDE := /usr/include BLAS_LIB := /usr/lib @@ -3,7 +3,7 @@ pkgname=caffe-cuda _srcname=caffe pkgver=1.0 -pkgrel=2 +pkgrel=3 pkgdesc='A deep learning framework made with expression, speed, and modularity in mind (with cuda)' arch=('x86_64') url='http://caffe.berkeleyvision.org/' @@ -29,7 +29,7 @@ conflicts=('caffe' 'caffe-git' 'caffe-cuda-git' 'caffe-cpu' 'caffe-cpu-git' source=("${pkgname}-${pkgver}.tar.gz"::"https://github.com/BVLC/caffe/archive/${pkgver}.tar.gz" 'Makefile.config') sha256sums=('71d3c9eb8a183150f965a465824d01fe82826c22505f7aa314f700ace03fa77f' - '9426828e7598bc85ae0ded53779643d99072e4e9c49f8a8b1deab8c166cbc2e7') + 'c61207ffea4ade927ce66a980f44732c1c2f9b82c500d51e687f5bcf773a4465') prepare() { cd "${_srcname}-${pkgver}" |