DarkNet 설치하는데 아래 그림 처럼 CUDA 버전이 11.1이라
버전에 맞게 CUDA랑 cudnn 설치하는 김에 매우 간단하게 포스팅도 함께 해보려고 합니다..(주륵)
DarkNet
C언어로 작성된 물체 인식 오픈 소스 신경망으로, DNN들을 학습시키고 실행시킬 수 있는 프레임워크입니다.
(Darknet-19: YOLOv2의 Network, Darknet-53: YOLOv3의 Network)
(+) YOLOv3(53)는 YOLOv2(19)에서 확장된 네트워크 구조를 가지는데,
복잡한 구조를 가져 속도가 느린 대신 작은 scale의 obejct에 대한 detect 성능이 증가합니다.
이 DarkNet을 사용하기 위해 호환되는 OpenCV, CUDA와 cudnn을 설치하겠습니다.
0-1. OpenCV 설치
https://www.learnopencv.com/install-opencv-4-on-windows/
(저는 사용하던 openCV가 있어서 따로 재설치 하지 않고 기존 것을 사용하겠습니다.)
0-2. CUDA 11.1 설치
https://en.wikipedia.org/wiki/CUDA#GPUs_supported
위 링크에서 자신이 사용하는 GPU가 해당 CUDA SDK version을 지원하는지 확인 후,
아래 링크를 통하여 CUDA Toolkit 11.1 설치
0-3. cudnn 설치
아래 링크에서 cudnn을 설치 후, 압축 풀기
https://developer.nvidia.com/cudnn
CUDA Toolkit이 설치된 경로(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1)에
압축 해제한 파일 복사 & 붙여넣기
0-4. nvcc -V 명령어로 설치된 CUDA Toolkit 버전 확인
1. vcpkg 설치
1.1 vcpkg 코드 다운로드
라이브 최상단으로 이동하여 이래 명령어로 vcpkg 코드를 다운로드합니다.
+) vcpkg란, Microsoft에서 Windows와 Linux, MacOS에서 C와 C++ 오픈 소스 라이브러리들의 자동 빌드 및 참조를 도와주는 패키지 관리자입니다.
1-2. vcpkg.bat 빌드
해당 코드가 다운로드 된 위치로 이동하여 폴더 안의 bootstrap-vcpkg.bat을 빌드합니다.
1-3. Visual studio 연동
1-4. 환경 변수 설정
시스템 속성 > 환경 변수에서 환경 변수 vcpkg를 추가합니다.
추가로, VCPKG_DEFAULT_TRIPLET라는 다른 시스템 변수를 정의하고 값을 x64-windows로 설정합니다.
Powershell 창을 열고, 다크넷 설치를 위한 vcpkg를 준비합니다.
1-5. vcpkg 빌드
+) 저는 다음과 같은 경고 문구와 함께 빌드 오류가 발생했습니다.
"Warning: The following VS instances are excluded beacuse the English Language pack is unavilable"
해결 방법:
Visual Studio 열기 → 도구 → 도구 및 기능 가져오기 → 언어팩 영어 선택 후 수정합니다.
2. Darknet 빌드
2-1. 프로젝트 다운로드
https://github.com/AlexeyAB/darknet
원하는 위치에 프로젝트를 다운 받은 후, 관리자 권한으로 PowerShell 실행합니다.
2-2. 프로젝트 위치로 이동 후, 다음 명령어로 Execution
+) Restricted가 뜨는 경우, 다음 명령어를 실행합니다.
2-3. 마지막으로 아래 명령어로 빌드를 합니다.
+) 만약, 오류가 뜬다면 CMake 설치 여부 및 환경 변수가 제대로 설정되어 있는지 확인해보시길 바랍니다.
마지막으로,
위 명령어를 통하여 darknet이 제대로 동작되는지 확인합니다.
<Reference>
https://medium.com/analytics-vidhya/installing-darknet-on-windows-462d84840e5a
https://int-i.github.io/cpp/2020-07-22/vcpkg-boost/
빌드 오류: https://github.com/microsoft/vcpkg/issues/10946
CMake 설치: https://jintan.tistory.com/49
'Deep Learning' 카테고리의 다른 글
[COCO] Ubuntu에서 COCO Dataset 다운 받기 (0) | 2021.02.25 |
---|