b__ono__ng

mmcv install error 본문

IT/Experience

mmcv install error

b__ono__ng 2024. 8. 24. 16:23

YOLO-World를 써보려고 했는데 mmcv 버전이 자꾸 안 맞는다고 한다.

pip로 계속 재설치해줬는데도 안 돼서 찾아보니, cuda 버전까지 맞춰줘야 한다고 한다...

그건 너무나도 귀찮은 일이라서 다른 방법을 찾아보니, 뭔가 cuda와 mmcv를 연결해주는 툴이 있나보다.

방법만 보면 아래와 같다.

pip install -U openmim
mim install "mmcv==2.0.0rc4"

 

참고 : https://mmcv.readthedocs.io/en/2.x/get_started/installation.html