Stargazer

[mmcv] RuntimeError: ms_deform_attn_impl_forward: implementation for device cuda:0 not found. 본문

인공지능/Issue

[mmcv] RuntimeError: ms_deform_attn_impl_forward: implementation for device cuda:0 not found.

COM2IT 2023. 4. 19. 01:02
반응형


이번에 mmcv가 개편이 되면서

내부 구조가 바뀌었고, 라이브러리 조차 변경점이 많았다.

그래서 자꾸 실행에 오류가 발생한다.

새로운 모델을 사용하고 싶어서

이 문제를 반드시 해결 해야만 했다.

보니까 mmcv 재설치를 진행하라고 한다.
그래서 버전을 명시해서 강제 설치를 진행했다.

mmcv 재설치하는 법

pip3 install --force  mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0.0/index.html

mmcv는 2.0.0버전을 사용할거라서 뒤에 cuda 버전과 torch 버전에 맞게 숫자 바꿔서 명령을 넣는다.

재설치하니까 드디어 에러가 안뜨고 작동을 시작했다
장장 6시간 만이다....

참조 :
https://github.com/open-mmlab/mmdetection/issues/6765

 

RuntimeError: nms_impl: implementation for device cuda:0 not found. · Issue #6765 · open-mmlab/mmdetection

When I run /usr/src/app/demo/inference_demo.ipynb, an error reported: /usr/src/app/mmdet/datasets/utils.py:65: UserWarning: "ImageToTensor" pipeline is replaced by "DefaultFormatBundle" for batch i...

github.com

https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html#id1

반응형

'인공지능 > Issue' 카테고리의 다른 글

[mmcv] ModuleNotFoundError: No module named 'projects'  (0) 2023.04.20
[CUDA] conda로 cuda 원하는 버전 설치하기  (0) 2023.04.19
[cuda] torch.cuda.is_available() false 문제 해결  (0) 2023.04.18
[pytorch] mmdetection 설치 방법  (0) 2023.04.05
[python] anaconda 삭제  (0) 2023.04.05
'인공지능/Issue' Related Articles more
Comments