[에러노트 / PANDAS] AttributeError: Can't get attribute '_unpickle_block' on <module 'pandas._libs.internals' from ' /internals.cpython-38-x86_64-linux-gnu.so'>

---

AttributeError : Can't get attribute '_unpickle_block' on 

<module 'pandas._libs.internals' from ' /internals.cpython-38-x86_64-linux-gnu.so'>

 

에러 메시지

---

AttributeError: Can't get attribute '_unpickle_block' on <module 'pandas._libs.internals' from ' /internals.cpython-38-x86_64-linux-gnu.so'>

 

에러 발생 지점

---

model = torch.load('/checkpoint/path')

 

해결법

---

checkpoint를 생성한 pandas 버전과 현재 읽어오려고 하는 local machine의 pandas의 minor version이 다를 때 발생하는 이슈이다. 

 

필자의 경우, 모델을 학습하는데에 사용했던 pandas는 1.4.1이었고, load 하려는 쪽에 깔려있던 pandas 버전은 1.3.1이었다. 버전을 1.4.1로 Upgrade 후, 문제가 해결되었다!

 

# Use Pip
pip install pandas==1.4.1

# Use Conda
conda install -c conda-forge pandas==1.4.1