Skip to content

Commit

Permalink
docker: specfic more limited set to build for cuda
Browse files Browse the repository at this point in the history
Signed-off-by: deadprogram <[email protected]>
  • Loading branch information
deadprogram committed Oct 10, 2023
1 parent 0c41dd6 commit 4b53132
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion Dockerfile.opencv-gpu-cuda-11
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ RUN curl -Lo opencv.zip ${OPENCV_FILE} && \
-D BUILD_opencv_cudacodec=OFF \
-D WITH_CUDNN=ON \
-D OPENCV_DNN_CUDA=ON \
-D CUDA_ARCH_BIN=5.2,6.0,6.1,7.0,7.5,8.0,8.6,8.9,9.0 \
-D CUDA_ARCH_BIN=6.0,6.1,7.0,7.5,8.0,8.6 \
-D OPENCV_GENERATE_PKGCONFIG=ON .. && \
make -j $(nproc --all) && \
make preinstall && make install && ldconfig && \
Expand Down
2 changes: 1 addition & 1 deletion Dockerfile.opencv-gpu-cuda-11.2.2
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ RUN curl -Lo opencv.zip ${OPENCV_FILE} && \
-D BUILD_opencv_cudacodec=OFF \
-D WITH_CUDNN=ON \
-D OPENCV_DNN_CUDA=ON \
-D CUDA_ARCH_BIN=5.2,6.0,6.1,7.0,7.5,8.0,8.6 \
-D CUDA_ARCH_BIN=6.0,6.1,7.0,7.5,8.0,8.6 \
-D OPENCV_GENERATE_PKGCONFIG=ON .. && \
make -j $(nproc --all) && \
make preinstall && make install && ldconfig && \
Expand Down
2 changes: 1 addition & 1 deletion Dockerfile.opencv-gpu-cuda-12
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ RUN curl -Lo opencv.zip ${OPENCV_FILE} && \
-D BUILD_opencv_cudacodec=OFF \
-D WITH_CUDNN=ON \
-D OPENCV_DNN_CUDA=ON \
-D CUDA_ARCH_BIN=5.2,6.0,6.1,7.0,7.5,8.0,8.6,8.9,9.0 \
-D CUDA_ARCH_BIN=6.0,6.1,7.0,7.5,8.0,8.6,8.9,9.0 \
-D OPENCV_GENERATE_PKGCONFIG=ON .. && \
make -j $(nproc --all) && \
make preinstall && make install && ldconfig && \
Expand Down

0 comments on commit 4b53132

Please sign in to comment.