평소처럼 모델 학습을 돌리려고 하는데, 너무 오래 걸리는 듯한 느낌이 들어 확인해보니까 현재 작업하고 있는 device가 cpu였다.
또 RuntimeError: The NVIDIA driver on your system is too old.라는 메시지도 출력되고 있었음..
찾아보니까 CUDA version과 Pytorch version이 호환이 되지 않아 발생하는 문제였다.
이럴 때는 보통 pytorch 사이트 내에서 확인해서 호환 가능한 pytorch 버전을 설치하면 되는데 내가 가진 서버의 경우, CUDA 11.4 version이라서 명확하게 나와 있진 않았다.
어떤 글을 보니까 CUDA 11.4 version을 쓸 때 pytorch 1.12.0 version을 쓰니 괜찮았다고 나와있어서 그렇게 깔아보니까 GPU를 정상적으로 쓸 수 있었다.
