공부공부공부를 합시다/이것저것 설치해 봅시다

[Ubuntu/Jupyter 서버 구축] Nvidia 설치 [2]

뻒음 2022. 8. 10. 15:37

tensorflow를 사용하려고 하면 Nvidia 그래픽 드라이버와 CUDA, Cudnn도 설치해줍니다.
저는 Nvidia 설치에서 난리가 났네요.. ㅎㅎ
설치 자체는 간단합니다.

1. 그래픽카드에 맞는 우분투 드라이버 탐색

sudo apt install ubuntu-drivers-common
ubuntu-drivers devices
model : GM200 [GeForce GTX TITAN X]
driver : nvidia-driver-515-server - distro non-free
driver : nvidia-driver-418-server - distro non-free
driver : nvidia-driver-510 - distro non-free
driver : nvidia-driver-510-server - distro non-free
driver : nvidia-driver-515 - distro non-free recommended
driver : nvidia-driver-470 - distro non-free
driver : nvidia-driver-390 - distro non-free
driver : nvidia-driver-450-server - distro non-free
driver : nvidia-driver-470-server - distro non-free
driver : xserver-xorg-video-nouveau - distro free builtin

저는 nvidia-driver-515를 추천한다네요.


2. 추천하는 버전으로 설치해줍니다.
설치 완료 후 reboot 해줘야 설치가 완료됩니다.

sudo apt install nvidia-driver-515
reboot


2.1. 나에게 발생한 오류
저는 드라이버를 설치하는 중에 이런 창이 떴습니다.
[Configuring Secure Boot] 라는 제목의 창이었는데요... 이때까지만 해도 저는 이게 문제의 원인인 줄 몰랐지...
하루하고도 반을 꼬박 날린 후에야 얘가 문제인 걸 알아버렸네요.

저는 BIOS에서 안전부팅모드, 빠른부팅모드를 비활성화한 후 설치하는 걸로 해결했습니다.
안전부팅모드, 빠른부팅모드 비활성화는 구글링하시기... 저는 매우 피폐한 상태로 작업해서 사진이 없답니다.
안전부팅모드 끄니까 부팅 속도도 엄청 빨라지더라구요

3. 설치 확인

nvidia-smi

 

이렇게 나오면 성공입니다. 만세!