일상 코딩
[windows 10/윈도우 10] darkflow 설치 및 환경설정 본문
설치 및 구동 환경
OS | WINDOWS 10 64bit |
CPU | i5-12400F |
GPU | NVIDIA GTX 1080 (Pascal 아키텍쳐) |
RAM | 64GB (이것보다 작아도 충분) |
설치 할 프로그램 목록 및 순서
1 | Visual Studio 2015 업데이트 3 |
2 | Anaconda |
3 | Nvidia 그래픽드라이버 |
4 | CUDA |
5 | cuDNN |
6 | Darkflow |
labelImg - 라벨링 툴
https://github.com/heartexlabs/labelImg/releases
Darkflow - YOLO 커스텀 학습 라이브러리
https://github.com/thtrieu/darkflow
Darkflow 설치 요구 버전
Tensorflow-gpu | 1.4.0 | 2017.11. |
그래픽드라이버 최소 버전 | 450.36.06+ 이후로 다 됨 | |
CUDA SDK 버전 | 8.0 | 2017.02. |
cuDNN | 6.0 | 2017.04.27. |
파이썬 | 3.6 | 2016.12.23. |
컴파일러 | Visual Studio 2015 업데이트 3 | 2015 |
빌드 도구 | CMAKE v3.6.3 |
Tensorflow-gpu - 1.4.0
https://www.tensorflow.org/install/source_windows#tested_build_configurations
nvidia 그래픽드라이버 최소 요구 버전
https://docs.nvidia.com/deploy/cuda-compatibility/#source-compatibility
Compute Capability에 따른 그래픽카드 및 CUDA 맞춤 버전
https://en.wikipedia.org/wiki/CUDA
풀칩을 사용하는 TITAN Xp(중고가격 30~40만원)를 중고로 사용하는게 구해서 시스템 구성하는걸 추천한다.
Visual Studio 2015 업데이트 3 Community iso 다운로드 설치
https://go.microsoft.com/fwlink/?LinkId=615448&clcid=0x409
https://www.kunal-chowdhury.com/2015/07/download-visualstudio-2015.html?m=0
Visual Studio 2022 Build Tools 설치
https://visualstudio.microsoft.com/ko/downloads/
다운로드 링크
https://aka.ms/vs/17/release/vs_BuildTools.exe
Anaconda 설치
https://www.anaconda.com/products/distribution
conda 가상환경 설정
관리자권한으로 PowerShell 실행
# 가상 환경 python 3.6버전으로 생성
conda create -n DFTest python=3.6 pip
# 가상 환경 진입
conda activate DFTest
# Tensorflow설치
pip install tensorflow-gpu==1.4
# 나머지 라이브러리 설치
pip install scipy numpy opencv-python
pip install cython
git 설치
https://git-scm.com/download/win
git을 통한 Darkflow 다운로드
cd ..
cd ..
git clone https://github.com/thtrieu/darkflow
PowerShell 창은 설치 종료시까지 끄지 않는다.
수동 다운로드
https://github.com/thtrieu/darkflow
darkflow는 이상태로 대기
그래픽드라이버 설치
https://www.nvidia.co.kr/Download/Find.aspx?lang=kr#
CUDA Toolkit 8.0 설치
https://developer.nvidia.com/cuda-toolkit-archive
위에서 자신에게 맞는 CUDA Toolkit 버전 및 그래픽카드를 찾았다면
이 단계를 진행한다.
https://developer.nvidia.com/cuda-80-ga2-download-archive
<설치시 유의사항>
사용자 정의 설치 > CUDA > Visual Studio Integration 체크 해제 후 설치한다.
cuDNN 6.0 설치
https://developer.nvidia.com/rdp/cudnn-archive
cuDNN 6.0 & CUDA 8.0 두개의 버전이 맞는지 확인 후 다운로드 한다.
cuDNN은 폴더안에 있는 파일을 옮기는 걸로 설치가 완료 된다.
방법은 아래 블로그로 가서 확인하면 자세히 나와있다.
Darkflow 빌드
# C 드라이브 최상단으로 위치에서 darkflow로 이동
# cd d까지 친후 Tab을 누르면 자동으로 darkflow가 완성된다.
cd darkflow
# build
python setup.py build_ext --inplace
입력 후 오류 메세지가 나오지 않으면 성공
python flow --h
입력 후 다음 메세지가 나오면 성공
중간 중간 다양한 오류가 나오게 될텐데
다음 참조 블로그를 보시면 될 것 같다.
참조
https://iagreebut.tistory.com/90
'딥러닝 > darkflow - YOLO' 카테고리의 다른 글
[Windows 10/darkflow] darkflow tiny YOLO V2 custom training command / tiny YOLO v2 커스텀 학습 방법 및 명령어 (0) | 2022.12.14 |
---|