From 77f70f8e9224ef5e6623ed026e34ddceed2cd890 Mon Sep 17 00:00:00 2001
From: Moritz Maxeiner <mm@ucw.sh>
Date: Tue, 10 Dec 2019 12:24:16 +0100
Subject: [PATCH] centos-7: mxnet: Limit cuda architectures to 6.1 (bin) and
3.0, 5.2, 7.5 (ptx)
---
centos-7/Dockerfile | 3 +++
1 file changed, 3 insertions(+)
diff --git a/centos-7/Dockerfile b/centos-7/Dockerfile
index 3d1d26f..9f85712 100644
--- a/centos-7/Dockerfile
+++ b/centos-7/Dockerfile
@@ -205,6 +205,9 @@ RUN version=1.5.1 && \
-D CMAKE_CXX_STANDARD=14 -D USE_CXX14_IF_AVAILABLE=ON \
-D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local \
-D USE_CUDA=ON -D USE_CUDNN=ON -D USE_NCCL=OFF \
+ -D CUDA_ARCH_NAME=Manual \
+ -D CUDA_ARCH_BIN="61" \
+ -D CUDA_ARCH_PTX="30 52 75" \
-D USE_OPENCV=ON -D USE_OPENMP=ON \
-D USE_LAPACK=ON -D USE_MKL_IF_AVAILABLE=OFF \
-D USE_PROFILER=ON \
--
GitLab