Ubuntu 16.04 Desktop에 nvidia driver 설치 삽질기

Source: Deep Learning on Medium


개인적으로 딥러닝 스터디를 하기위해 GTX 1070을 탑재한 게이밍 노트북을 중고로 구매하여 세팅을 하다가 nvidia driver 설치에서 난항을 겪었던 삽질기를 공유하고자 합니다.

참고로 제 장비는 msi GT73VR 7RE Titan 모델입니다. 
다른건 중요하지 않고 GTX 1070 8GB가 있는 모델인데 1TB의 하드디스크에서 500GB를 Ubuntu 설치를 위해 파티션을 나누고 Ubuntu 16.04 Desktop을 설치했습니다.

nvidia driver를 설치하는 예시를 좀 찾아봤는데 이상하게도 로그인 무한루프 라는 용어가 많이 보이더군요. 왜 무한 루프가 돌지?라고 생각했는데 그냥 쉽게 설치가 될걸로 생각했던 nvidia driver 설치가 쉽진 않았습니다.

삽질1

NVIDIA 공식 홈페이지에서 Linux용 드라이버를 다운받아 가이드에 따라 설치를 진행했습니다. 설치 완료 후 재부팅을 하면 로그인 화면에서 로그인을 해도 다시 로그인 화면이 뜹니다. 원인은 잘 모르겠습니다.

삽질2

그렇다면 Ubuntu Server를 설치해야 하나? 설명에 Server버전에 설치하는게 많던데라고 생각하며 Server를 설치해봅니다. 실패했습니다. 
네트워크 인터페이스를 찾지 못하고 Archive Mirror 사이트를 찾지 못해 진행이 되지 않았습니다.

삽질3

특정 프로그램으로 부팅 디스크를 만들면 위 단계를 패스할 수 있다는 글을 보고 재도전 합니다. Ubuntu Server는 어떻게 설치 했는데 WiFi를 설정할수가 없었습니다. 콘솔만 있으니 더 어려웠고 WiFi를 잡는 커맨드는 apt-get으로 설치를 해야한다고 나옵니다. Desktop으로 돌아가기로 합니다.

해결

다시 Ubuntu Desktop으로 돌아오고 login loop 이라는 키워드와 함께 구글링을 해봤습니다. 아래와 같은 3개의 커맨드를 얻었습니다. (https://askubuntu.com/a/848668)

sudo ubuntu-drivers list

이러면 설치가 가능한 nvidia driver 가 출력됩니다.
설치해봤습니다.

sudo ubuntu-drivers auto install

순조롭게 설치가 됐습니다.
설명대로 마무리 커맨드를 넣습니다.

sudo service lighted restart

로그인을 해봅니다.
그런데 뭔가 이상하게 검은 화면만 나오고 진행이 되지 않았습니다.
재부팅을 해보니 다행히 정상 동작합니다.

마지막으로 드라이버를 확인해봅니다.

nvidia-smi

드디어 정상 설치된 정보가 출력됩니다. 감격 ㅠㅠ

딥러닝 입문하시는 분들 중에 nvidia 드라이버 설치에 난항을 겪으시는 분들이 계시다면 위 커맨드를 참고해보세요.