tensorflow openCV with GPU
- 설치 환경GPU - NVIDIA GeForce RTX 3060tensorflow 2.9.1 → 2.7.0
- tensorflow-gpu 2.7.0
- python 3.9.12
- OS - windows10 x64
1. NVIDIA 그래픽 드라이버 설치
https://www.nvidia.co.kr/Download/index.aspx?lang=kr
설치 후 cmd 창에서 nvidia-smi해서 드라이버 선택됐는지 확인
2. CUDA Toolkit 설치
- Cuda Toolkit 버전 확인 - Release Notes :: CUDA Toolkit Documentation
- Cuda Toolkit 아카이브 - CUDA Toolkit Archive
→ CUDA 11.2.0 다운로드
3. cuDNN 설치
- cuDNN과 tensorflow, CUDA 호환 버전 확인 - Build from source | TensorFlow
- cuDNN 아카이브 - cuDNN Archive
→ cuDNN 8.1.0 다운로드
cuDNN 압축 해제하고 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2에 덮어쓰기
nvcc —version 입력으로 확인
4. tensorflow 연동 테스트
GPU 잘 잡히는지 확인
5. openCV CMake 및 빌드, 환경변수 설정
4.5.1 버전으로 zip 파일 다운로드
- openCV - Release OpenCV 4.5.1 · opencv/opencv
- openCV-contrib - Release 4.5.1 · opencv/opencv_contrib
다운로드 후
[CUDA&OpenCV] Python 환경 OpenCV GPU 연동 (CUDA backend in OpenCV) - 2. 빌드 & 테스트 참고하여 빌드
Visual Studio 2022에선 오류 발생
Visual Studio 2017에서 CMake 및 빌드 해야됨
CMake, Visual Studio 2017 둘 다 관리자 권한으로 해야함!
6. 테스트
