엔비디아 GPU 활용을 위한 드라이버 설치(우분투 22.04 기준)

 

Stable diffusion 과 같은 생성 AI 프로그램은, 외장 그래픽 카드(GPU)의 자원을 활용하여 이미지를 생성합니다.
즉, Stable diffusion을 이용해 AI 이미지를 생성하는데에는 GPU가 있어야 하고, 그 성능도 좋아야 한다는 말이죠.

일반적으로 생성 AI를 위한 외장 그래픽 카드는 Nvidia 사의 그래픽 카드를 많이 사용합니다.

(Nvidia가 생성 AI와 관련한 라이브러리 및 패키지를 잘 개발해서 제공하고 있기 때문이죠)

 

여기서는 우분투 os 환경에서 Nvidia 그래픽 카드(이하 GPU)를 활용하여 생성 AI 작업을 하기 위한 선행 작업으로서 Nvidia GPU 드라이버 설치 방법에 대해 알려드리겠습니다.

(GPU를 정상적으로 사용하기 위해서는 GPU 드라이버가 설치되어 있어야 그 자원을 사용할 수 있습니다!)

 

1. GPU 드라이버 설치

먼저, 우분투 패키지 업데이트를 진행합니다.

 

최신 드라이버를 설치할 때는 현재 우분투에 설치된 패키지들이나 커널이 최신 상태를 유지해야 설치 시 충돌이 발생하지 않으므로, 항상 패키지 업데이트 작업을 선행해야 합니다.

 

우분투 터미널(terminal) 창을 열고 아래 명령어를 복사해서 입력해 주세요.

sudo apt update && sudo apt upgrade -y
sudo apt install build-essential

 

위 작업이 완료되었으면 다음 명령어를 입력합니다.

ubuntu-drivers devices

 

현재 연결된 그래픽 카드의 정보와 추천하는 driver 목록들이 나열됩니다.

이 중에서 원하는 버전을 설치해야 하는데, 일반적으로 recommand 라고 뒤에 붙은 버전을 설치하면 됩니다.

(단, 드라이버 뒤에 server 라고 붙은 것은 서버용 버전이므로 일반 버전으로 설치하시면 됩니다.)

 

ubuntu-drivers devices 명령어 출력 결과 예시

 

저의 경우에는 server 용이 아닌 버전 중에 470, 525, 535 open 버전을 설치하면 된다고 나오네요.

목록에서 설치할 버전을 고르셨으면 다음과 같은 명령어로 설치를 진행합니다.

 

sudo apt install nvidia-driver-원하는숫자

 

예를 들어 nvidia-driver-525-open 설치를 원하면 다음과 같은 명령어를 사용

sudo apt install nvidia-driver-525-open

 

설치 작업이 완료되면 컴퓨터 재시작을 합니다.

다시 부팅이 완료되면, 정상적으로 그래픽 카드 드라이버가 설치되었는지 다음의 명령어로 확인합니다.

 

반응형

 

nvidia-smi

 

정상적으로 설치가 된 경우, 다음의 명령어가 정상 작동하고, 아래와 같이 드라이버 정보와 사용량 등이 출력됩니다.

 

 

 

만약에 해당 명령어가 작동하지 않는다면 정상적으로 설치되지 않은 것이니, 

Clean 삭제 절차를 거친 후 다시 설치 작업을 진행해 주세요.

 

Clean 삭제 방법(아래 명령어를 모두 복사해서 실행하시면 됩니다.)

sudo apt-get remove --purge 'nvidia-.*' -y
sudo apt-get --purge remove 'cuda*'
sudo apt-get autoremove --purge 'cuda*' -y

sudo rm -rf /usr/local/cuda