어떨게 이런 오류까지....올리나 싶지만
난 기억력이 개똥이라구ㅠ
이건 너무너무너무 간단한 오류..
torch pre-trained 모델 돌리던 중 cuda 문제로 난 오류인데
torch.device("cuda:o" if torch.cuda.is_available() else "cpu")
model.load_state_dict(torch.load("model.pt"))
이 코드 부분에서 cuda / nvidia driver 버전이랑 pytorch 버전이 맞기 않아서 생기는 문제 같음
cuda를 torch 버전에 맞춰 재설치 해주면 되는데 cuda 버전 변경이 어렵다면 torch 버전을 cuda에 맞춰 변경 해줘도 되고
이것도 귀찮고 cpu 사용해도 성능이나 속도 문제가 상관없다 하면 그냥
model.load_state_dict(torch.load("model.pt", map_location='cpu'))
로 코드 변경해줘도 잘 돌아감~
이 오류는 dlib 설치 문제인지 face_recognition 설치 중에 오류가 발생했다는데
이유는 정확하게 모르겠음...
일단 해결방법은
pip3 uninstall dlib
pip3 uninstall face_recognition
pip uninstall dlib
pip uninstall face_recognition
pip3 install -v --install-option="--no" --install-option="DLIB_USE_CUDA" dlib
pip3 install face_recognition
dlib이랑 face_recognition 삭제하고 위에 코드대로 재설치하면 오류 해결!
다른 또 하찮은 오류는 이 게시물에 계속 수정해서 추가할 예정임~
얼렁뚱땅 마무리쓰~
얼렁뚱땅