diff --git a/centos-7/Dockerfile b/centos-7/Dockerfile index 7f854e4bc58cee0f5f1c8ba50e14941d0b919390..17079d4aa726fb6bc75701fc38b22f80a6a7af8a 100644 --- a/centos-7/Dockerfile +++ b/centos-7/Dockerfile @@ -168,6 +168,8 @@ RUN version=3.4.8 && \ -D WITH_IPP=OFF \ -D WITH_CUDA=ON \ -D WITH_CUBLAS=ON \ + -D CUDA_ARCH_BIN="61" \ + -D CUDA_ARCH_PTX="30 52 75" \ -D BUILD_opencv_cudacodec=OFF \ -D WITH_EIGEN=ON \ -D WITH_FFMPEG=ON \