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)