DL

YOLOV5 install for mac

Haribo- 2022. 11. 22. 17:49

YOLO v5 Pytorch 깃헙 링크 : https://github.com/ultralytics/yolov5

 

1. 가상환경 생성

conda create -n yolov5 python=3.8
conda activate yolov5

2.  깃헙 클론 

#code clone
git clone https://github.com/ultralytics/yolov5.git

3. 필요한 라이브러리 설치

yolo v5 깃헙의 requirements.txt 참고해서 설치 

 

# update
conda update -yn base -c defaults conda

# install Lib for YOLOv5
conda install -c anaconda cython numpy pillow scipy seaborn pandas requests
conda install -c conda-forge matplotlib pyyaml tensorboard tqdm opencv 

#install pytorch
#자신 pc에 맞는 파이토치 깔기 
conda install pytorch torchvision torchaudio -c pytorch
#conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

# Extra
conda install -c conda-forge onnx

자신의 pc에 맞는 torch install version 체크 가능 

https://pytorch.org/get-started/locally/