diff --git a/centos-7/Dockerfile b/centos-7/Dockerfile index 3d1d26fc9f97b6d13a1cb530bf59a38e9b1f5e91..9f857122a694574ecc53237bc522ee0c528fbdce 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 \