RTX3090 & Python3.6으로 YOLOv7 실행하기
가지고 있는 GPU와 필요한 Python 버전을 잘 고려해서 설치해야 합니다.
기본 가지고 있는 것
GPU: RTX 3090
Python: 3.6.9
먼저 설치해야 하는 것
CUDA: 11.3
cuDNN: 8.9.4
아래 링크를 참고해서 설치합니다.
Ubuntu 18.04에서 CUDA, Cudnn 설치
데스크탑에서 Yolo를 사용하려면 CUDA와 Cudnn을 설치해줘야 합니다. Jetson은 Jetpack을 설치할 때 알아서 다 깔리지만 데스크탑은 따로 설치를 해줘야 해서 좀 귀찮.. 먼저 데스크탑에 이미 설치된 옛
kumoh-irl.tistory.com
YOLOv7에 필요한 라이브러리 설치
먼저 yolov7 깃허브를 참고해 requirement.txt를 이용해 모두 깔아줍니다.
GitHub - WongKinYiu/yolov7: Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time
Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors - GitHub - WongKinYiu/yolov7: Implementation of paper - YOLOv7: Trainable bag-of...
github.com
그리고 pytorch의 버전을 1.10+cu102에서 1.7+cu110으로 낮춰줘야 합니다.
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html