Computer Science, Programming/Git, Docker, Server, Linux
GPU 서버 사용법 - CUDA, PyTorch 버전 맞추기 (총정리)
논문 구현을 해볼 때마다 PyTorch버전에 따라 필요한 CUDA 버전이 다르고, 버전이 서로 맞지 않아 시간을 낭비하는 경우가 많았다. 이를 해결하기 위해 (내가 썼던..) 여러 글을 참조해서 docker 컨테이너를 만들어줬는데, 과정을 축약해서 하나의 글로 정리해보려 한다. 이미지 빌드, 컨테이너 생성 등의 긴 과정은 생략하고, 간단하게 필요한 PyTorch 버전에 따라 CUDA 버전을 찾는 과정 까지를 다룬다. 서버에서 맞는 버전을 찾고, 컨테이너를 생성한 후에 설정까지 해주는 전체 과정은 다음 글을 참조하자. (링크 추후에 추가) 목차 Finding Version 전체 과정을 요약하면 다음과 같다. nvidia-smi로 gpu 정보 얻기 (모델명, driver 버전 등) Driver version..
2022. 10. 18. 20:09
최근댓글