본문 바로가기

컴퓨터

Docker 내의 pytorch 이미지 컨테이너에서 GPU 인식 문제(NVIDIA 그래픽 드라이버 문제)

몇 달 전에 테스트한 도커를 이용한 프로젝트가 갑자기 작동을 하지 않게 되었는데 설정이 꼬여서 그런가 싶어서 2일 동안 포맷 설정을 반복했는데 드라이버 문제일지도 모른다는 깃허브 글을 보고 테스트했던 시기의 드라이버로 변경하니 자연스럽게 해결되었습니다.

NVIDIA 그래픽 드라이버에 대해서는 잘 몰랐었는데 이렇게 버전이 나눠져 있기에 권장 버전이나 인증된 버전을 선택하는 것이 더 안전해보여서 권장 드라이버로 설치했습니다. 저와 같이 고통받고 있다면 한 번 시도해보는 것도 좋을 것 같습니다.

https://www.nvidia.co.kr/Download/Find.aspx?lang=kr

 

Advanced Driver Search official NVIDIA drivers

Advanced Driver Search official NVIDIA drivers

www.nvidia.com