From c19313c7dc0ac8882db274e74999d3cbf91e74f3 Mon Sep 17 00:00:00 2001 From: osehyeon Date: Sun, 2 Feb 2025 03:12:02 +0900 Subject: [PATCH] add: cuda build --- requirements.txt | 1 + setup.sh | 11 +++++++++++ 2 files changed, 12 insertions(+) create mode 100644 requirements.txt create mode 100644 setup.sh diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 00000000..1477fc8e --- /dev/null +++ b/requirements.txt @@ -0,0 +1 @@ +scikit-build \ No newline at end of file diff --git a/setup.sh b/setup.sh new file mode 100644 index 00000000..d7ca64cb --- /dev/null +++ b/setup.sh @@ -0,0 +1,11 @@ +#!/bin/bash + +ENABLE_CONTRIB=1 python setup.py bdist_wheel -- \ + -DWITH_CUDA=ON \ + -DWITH_CUDNN=ON \ + -DOPENCV_DNN_CUDA=ON \ + -DENABLE_FAST_MATH=1 \ + -DCUDA_FAST_MATH=1 \ + -DWITH_CUBLAS=ON \ + -DCUDA_ARCH_BIN=6.1 -- \ + -j $(nproc)