상황 설명
HMR(human mesh recovery)의 코드를 실행하는 도중
tensorflow-gpu 1.3.0 버전을 설치하고
python -m demo --img_path data/coco1.png를 실행하니
ImportError: libcublas.so.x.0 cannot open shared object file: No such file or directory
에러가 발생했습니다!
https://github.com/akanazawa/hmr
akanazawa/hmr
Project page for End-to-end Recovery of Human Shape and Pose - akanazawa/hmr
github.com
에러 설명
해당 에러는 CUDA 버전과 관련이 있는 것 같습니다.
tensorflow-gpu 1.3.0 버전에는 libcublas.so.8.0 에러가 발생하고
tensorflow-gpu 1.5.0 버전에는 libcublas.so.9.0 에러가 발생합니다.
저의 경우 cuda 10.2 버전을 이용중입니다.
CUDA 버전 확인 방법
cuda 버전 확인 방법은 다음과 같습니다.
$nvidia-smi
그럼 위와 같이 우측 상단에 CUDA 버전이 나옵니다.
만약, nvidia-smi이 실행되지 않는다면 nvidia driver 설치를 해보세요!
https://grow-up-by-coding.tistory.com/2
ndivia driver 설치(우분투20.04 , 18.04, 16.04)
우분투를 처음 시작하면 ndivia가 설치 되어 있지 않습니다. 이를 해결하기 위해 아래의 블로그를 참고하여 해결하시길 바랍니다! 20.04버전 https://finding-in-daily.tistory.com/163 18.04버전, 16.04버전 http..
grow-up-by-coding.tistory.com
해결 방법
pip install tensorflow-gpu ==1.12.0을 설치하면 해당 에러가 사라질 것입니다!
1.12.0 버전은 CUDA 10버전을 지원하는 것 같아요!
<수정>
tensorflow-gpu 버전이 문제가 아니었습니다!
우선 hmr github에 보면 demo를 실행하기 위해 1.12.0버전을 사용하는 것은 맞습니다.
하지만 해당 에러는 tensorflow-gpu가 아닌 tensorflow 1.12.0버전을 설치하면 해결이 됩니다!
결론은
$ pip install tensorflow-gpu ==1.12.0
$ pip install tensorflow==1.12.0
둘 다 하시면 됩니다!
