[Error] ModuleNotFoundError: No module named 'mmcv.

Skills/Error

Suyeon Cha 2024. 1. 18. 02:54

728x90

segmentation이나 detection의 task를 진행할 때, 많이 사용되는 라이브러리 mmedetection이 있습니다. 이 라이브러리에서 사실 에러가 많이 나는데 ... 프로젝트 깃헙의 가이드에 따라 해결이 잘 될 때도 안 될 때도 있는 것 같습니다.

그중 mmcv._ext가 안 된다는 에러가 발생했는데,

pip uninstall mmcv
pip install mmcv-full

저는 운 좋게도 이 방법으로 해결했습니다. 그런데 오류를 검색하고 깃헙 이슈를 보면

너무 최악의 피드백 ,,,

하지만 오류의 이유로는 'mmcv-full'이 설치되지 않아서 인듯하여

pip install mmcv-full

설치를 한 번 진행해보시면 좋을 것 같습니다.

그리고 원인은 pytorch, cuda, mmcv의 3박자가 맞아 떨어져야 하는 것 같습니다. 따라서 최대한 깃헙에 명시되어 있는 pytorch 버전을 먼저 맞게 설치를 해주시고,

이곳에서 cuda와 torch 버전에 맞는 것을 찾아 아래의 예시처럼 설치해주세용~!

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html

이렇게 뜨겁고 긴 이슈는 처음 보는 것 같습니다 ㅎㅎ

728x90