ModuleNotFoundError: No module named 'mmcv._ext'

 

mmcv-full과 mmsegmentation 설치를 할때 아래와 같이 진행했다.

pip install mmcv-full==1.3.0 mmsegmentation==0.11.0

 

 

그런데 train 코드를 실행하는데 다음과 같은 오류가 떴다.

ModuleNotFoundError: No module named 'mmcv._ext'

 

서치를 해본 결과 mmcv를 uninstall 하고 다시 mmcv-full을 설치하여 해결했다는 글을 보고

pip uninstall mmcv를 했지만 warning 문구가 뜨면서 uninstall이 안되었다.

 

그래서 그냥 가상환경을 새로 만들고 torch 설치부터 mmcv, mmsegmentation을 다시 설치하기로 했다. 

mmcv 공식 문서를 보니 installation instructio을 보고 제대로 설치를 해야겠다고 생각이 들었다. 

https://mmdetection.readthedocs.io/en/v2.11.0/faq.html#mmcv-installation

 

나의 CUDA 버전, torch 버전을 고려했을 때 mmcv 2.2.0을 설치해야 한다고 한다.

https://mmcv.readthedocs.io/en/latest/get_started/installation.html#install-with-pip

 

내가 따라가고 있는 github repo에서는 mmcv-full==1.3.0과 mmsegmentation==0.11.0을 사용하고 있어서 호환성이 맞을지는 모르겠지만 일단 시도해보자

 

흠.. 또 오류가 떴다.

ModuleNotFoundError: No module named 'mmcv.fileio'

 

뭐 어쩌라는건지!!

 

---

repo와 똑같이 맞춰주기 위해서 mmcv-full==1.3.0과 mmsegmentation==0.11.0을 설치해주는 것이 베스트다.

 

알고보니 mmcv-full 2미만 버전은 PyTorch 1.x.0 버전에서만 컴파일된다고 한다.

나는 cuda 12.1과 torch 2.3.0 버전을 가지고 있기 때문에 mmcv-full 1.3.0 와 호환이 안된다.

(설치는 되었지만 컴파일에서 오류가 나는 상황)

 

해법은 torch 버전을 낮추든.. cuda 버전을 낮추든 해야된다는 것

 

아 귀찮아

 

 

 

 

(수정)

결론: 그냥 도커 쓰세요..