가지고 있는 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
'정나우 > 코드' 카테고리의 다른 글
| 우분투 터미널 안 열림 & Xterm 없을 때 (0) | 2023.08.22 |
|---|---|
| 우분투 18.04 python3 버전만 바꾸기 (0) | 2023.08.22 |
| 무신사 재입고 알람 프로그램 만들기 (0) | 2023.04.12 |
| 파이썬, pip 설치 오류 (0) | 2023.04.12 |
| Opencr로 다이나믹셀(Dynamixel) 제어하기 (1) | 2022.11.07 |
댓글