[Python] ubuntu 20.04에서 venv에서 OpenCV 연동하기

ubuntu 20.04에서 파이썬 버전을 확인해보면 python 3.8이 설치되어 있습니다.

hunature@hunature-ThinkStation-P500:~$ python3 --version
Python 3.8.10

파이썬의 pip와 venv를 사용하기 위하여 다음과 같이 패키지들을 설치합니다.

hunature@hunature-ThinkStation-P500:~$ sudo apt install python3-pip python3.8-venv

다음과 같이 venv 모듈을 이용하여 vpc를 위한 가상환경을 생성하고, 활성화합니다.

hunature@hunature-ThinkStation-P500:~$ mkdir -p ~/work/venv
hunature@hunature-ThinkStation-P500:~$ cd ~/work/venv
hunature@hunature-ThinkStation-P500:~/work/venv$ python3 -m venv vpc
hunature@hunature-ThinkStation-P500:~/work/venv$ cd vpc/bin
hunature@hunature-ThinkStation-P500:~/work/venv/vpc/bin$ source activate
(vpc) hunature@hunature-ThinkStation-P500:~/work/venv/vpc/bin$ 

pip를 이용하여 "opencv-python" 패키지를 설치합니다.

(vpc) hunature@hunature-ThinkStation-P500:~$ pip install opencv-python
Collecting opencv-python
  Downloading opencv_python-4.5.5.64-cp36-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (60.5 MB)
     |████████████████████████████████| 60.5 MB 14.9 MB/s
Collecting numpy>=1.17.3; python_version >= "3.8"
  Downloading numpy-1.22.4-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (16.9 MB)
     |████████████████████████████████| 16.9 MB 75.5 MB/s
Installing collected packages: numpy, opencv-python
Successfully installed numpy-1.22.4 opencv-python-4.5.5.64

이제 OpenCV 소스를 동작할 모든 준비가 완료되었습니다.

 

OpenCV 관련 소스를 실행해 보면 됩니다.

# OpenCV 정보 확인 예제
import cv2

# 패키지 설치 위치 확인
print(cv2.__file__)

# 패키지 버전 확인
print(cv2.__version__)

# 쿠다를 사용할 수 있는 장치 수
print(cv2.cuda.getCudaEnabledDeviceCount())

위 예제를 실행한 결과는 다음과 같습니다.

(vpc) hunature@hunature-ThinkStation-P500:~/work$ python cv_info.py
/home/hunature/work/venv/vpc/lib/python3.8/site-packages/cv2/__init__.py
4.5.5
0

위와 같이 최신의 OpenCV가 설치되어 사용 가능하지만, DNN의 경우에는 CUDA는 활용할 수 없고 CPU만으로 처리됩니다.

 

CUDA를 활용하기 위해서는 직접 빌드하여 설치해야만하는데, 이 부분은 다음에 정리해 보겠습니다.