mmpose 소스코드 실행시 아래와 같은 오류가 발생하여 확인해보니 mmcv 라이브러리와 토치 버전이 호환되지 않아 발생한 현상입니다.
지속적으로 토치 버전을 다운그레이드해서 해결했습니다.
최신 버전은 호환되지 않으며 버전 1.11.0과 호환됩니다.
오류 기록
ImportError: ~/anaconda3/mmpose/lib/python3.8/site-packages/mmcv/_ext.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZNK3c1010TensorImpl36is_contiguous_nondefault_policy_implENS_12MemoryFormatE
솔루션 (cuda 버전에 따라 사용 가능)
cuda 11.3의 경우 아래 명령어를, 그 외 버전의 경우 아래 링크를 참조하세요.
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
파이토치 설치 링크
https://pytorch.org/get-started/previous-versions/#v1110
토치
연구 프로토타이핑에서 프로덕션 배포까지의 프로세스를 가속화하는 오픈 소스 머신 러닝 프레임워크입니다.
pytorch.org