diff options
| author | Jingbei Li | 2020-07-31 23:19:35 +0800 |
|---|---|---|
| committer | Jingbei Li | 2020-07-31 23:19:35 +0800 |
| commit | 430f7ddf106531fdf0e94ccaf42485aafb42e816 (patch) | |
| tree | ab7e909d5b8e4d934c8a395c54b6e1f044d1452f /PKGBUILD | |
| parent | d84d734be70559f9b3ad858b896397ad520819d8 (diff) | |
| download | aur-430f7ddf106531fdf0e94ccaf42485aafb42e816.tar.gz | |
upgraded to 1.7.0.rc1
Diffstat (limited to 'PKGBUILD')
| -rw-r--r-- | PKGBUILD | 5 |
1 files changed, 4 insertions, 1 deletions
@@ -2,7 +2,7 @@ # Contributor: Carl Åkerlindh <carl.akerlindh at gmail dot com> pkgbase=mxnet pkgname=('mxnet' 'mxnet-cuda' 'mxnet-mkl' 'mxnet-mkl-cuda') -pkgver=1.7.0.rc0 +pkgver=1.7.0.rc1 pkgrel=1 pkgdesc="Flexible and Efficient Library for Deep Learning" arch=('x86_64') @@ -104,6 +104,8 @@ prepare() { cd "$srcdir/$pkgbase" git submodule update --init --recursive + rm -rf 3rdparty/nvidia_cub + # https://github.com/apache/incubator-mxnet/pull/18357 git checkout 78e31d6 src/operator/tensor/elemwise_binary_broadcast_op_basic.cc @@ -150,6 +152,7 @@ build() { cuda_args=( -DUSE_OPENCV:BOOL=OFF -DUSE_NCCL:BOOL=ON + -DMXNET_CUDA_ARCH='5.0;6.0;7.0;7.5;8.0' -DCMAKE_C_COMPILER=/opt/cuda/bin/gcc -DCMAKE_CXX_COMPILER=/opt/cuda/bin/g++ ) |