(참고)cuda version 10.1 기준으로 설치를 했습니다
- Detectron2의 공식 버전은 windows에 지원하지 않기 때문에 해당 설치 방법을 따라하시면 완벽하게 작동이 됩니다
Anaconda Prompt에서 가상환경을 생성하고 접속
(conda create -n 가상환경이름 python=파이썬버전)
(conda activate 가상환경이름)
conda create -n detectron2 python=3.7
conda activate detectron2
cuda 설치
- pythorch버전 1.6, 1.5, 1.4와 함께 사용할 수 있는 cuda 10.1을 사용합니다
https://developer.nvidia.com/cuda-10.1-download-archive-base
CUDA Toolkit 10.1 original Archive
Select Target Platform Click on the green buttons that describe your target platform. Only supported platforms will be shown. Operating System Architecture Distribution Version Installer Type Do you want to cross-compile? Yes No Select Host Platform Click
developer.nvidia.com
Pytorch 설치
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch
- pytorch 1.7 이상은 작동하지 않습니다
필요한 라이브러리 설치
pip install cython
pip install "git+https://github.com/philferriere/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI"
Detectron2 설치
git clone https://github.com/DGMaxime/detectron2-windows.git
- windows에서 작동할 수 있도록 일부 파일이 수정 되었습니다
setup.py
detectron2/engine/defaults.py
detectron2/layers/csrc/cocoeval/cocoeval.cpp
Detectron2의 설치가 완료되면 폴더로 이동해준 다음 라이브러리를 설치합니다
cd detectron2-windows
pip install -e .
pip install opencv-python
- 설치에 성공하면 “Successfully installed detectron2” 이란 메시지가 나와야 합니다.
테스트
python tests/test_windows_install.py
- 다음과 같이 실행됬다면 성공입니다!
!주의 : 실행시 오류가 뜬다면 라이브러리 설치나 CUDA를 제대로 인식 하지 못할 경우가 드물게 있습니다. 그럴경우에는 라이브러리를 지우고 다시설치해보시고 그래도 안된다면 가상환경을 완전히 지운 다음에 CUDA를 재설치 후 다시 가상환경을 만들어 보시기 바랍니다
- 가상환경 삭제
(conda remove --name 가상환경이름 --all)
conda remove --name detectron2 --all
- 해당글을 참고하였습니다
https://dgmaxime.medium.com/how-to-easily-install-detectron2-on-windows-10-39186139101c