(참고)cuda version 10.1 기준으로 설치를 했습니다
Anaconda Prompt에서 가상환경을 생성하고 접속
(conda create -n 가상환경이름 python=파이썬버전)
(conda activate 가상환경이름)
conda create -n yolov5 python=3.8
conda activate yolov5
yolov5 설치하고 yolov5폴더로 이동
(cd yolov5를 설치하고자 하는 폴더위치)
(git clone 깃허브주소)
cd C:/
git clone https://github.com/ultralytics/yolov5.git
cd yolov5
내 cuda version을 확인
nvcc --version
- cuda가 설치되어 있지 않은 경우에는 밑에 주소에서 다운 받으시면 됩니다 (다운 받고나서는 anaconda prompt를 재부팅)
- 저는 10.1을 사용하기에 10.1 버전기준으로 알려드리겠습니다
https://developer.nvidia.com/cuda-10.1-download-archive-base
설치 (cuda10.1)
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch!주의 : cudatoolkit은 반드시 내 cuda 버전과 같아야 합니다
필요한 라이브러리들을 설치
pip install -r requirements.txt
!주의 : 라이브러리는 yolov5와 나의 가상환경 설정에 따라 보이는 것과 버전은 다를 수도 있습니다
- 정말 드물게 라이브러리를 한번에 받다가 오류가 생기는 경우가 있는데 yolov5 tutorial을 실행할 때 오류가 생긴다면 먼저는 오류가 뜨는 라이브러리를 다시 설치해보고 그래도 안된다면 완전히 다 지웠다가 가상환경부터 다시 설치해보고 라이브러리들도 하나씩 설치해보면 해결되기도 합니다
테스트
python detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source data/images- yolov5폴더 안에 run/detect/exp 폴더안에 밑에 처럼 사진에 객체감지(object detection)가 된다면 성공입니다
다음에는 custom data를 활용한 훈련과 결과에 대해서 소개드리겠습니다