OpenCV-Python-GPU

Roboflow에서 받은 데이터 Deeplab학습을 위한 라벨 형식 변경 작업 중
for문은 pixel 체크에서 느린 작업 속도 체감
속도 향상을 위해 cuda 도입 중 conda 환경에서 cuda 인식 안됨
conda 환경에서 opencv-python gpu 버전 설치하기

Install path

<아나콘다_설치_경로>/envs/<가상환경_이름>/lib/python<버전>/site-packages/

git clone --recursive https://github.com/opencv/opencv-python.git

CUDA 호환 버전 확인

cd /usr/local/cuda/samples
make
./bin/x86_64/linux/release/deviceQuery

CUDA Capability Major/Minor version number 8.6

Build

-DCUDA_ARCH_BIN은 본인 GPU's arch에 맞게 세팅

ENABLE_CONTRIB=1 python setup.py bdist_wheel -- \
	-DWITH_CUDA=ON \
	-DWITH_CUDDN=ON \
	-DOPENCV_DNN_CUDA=ON \
	-DENABLE_FAST_MATH=1 \
	-DCUDA_FAST_MATH=1 \
	-DWITH_CUBLAS=1 \
	-DCUDA_ARCH_BIN=<CUDA Capability version number> -- \
	-j $(nproc)

sudo apt upgrade -y
sudo apt reinstall libffi-dev
export LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libffi.so.7
pip install albumentations opencv-python-headless
cd dist
python -m pip install --upgrade \
	   --force-reinstall \
       opencv_contrib_python-<version>-cp<version>-cp<version>m-linux_x86_64.whl

Result

성공은 했지만 인식 문제와 결과 처리 문제 등으로 torch cuda 사용으로 결정