mmpose 소스 실행 중 아래와 같은 오류가 발생해서 확인해보니, mmcv 라이브러리와 torch 버전이 호환되지 않아 발생한 문제였습니다.
torch 버전을 계속 다운그레이드 해서 해결했습니다.
최신 버전은 호환이 안되고, 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
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.org