Ubuntu에 CUDA 설치

Ubuntu에 CUDA 설치

 

이전 포스팅에서는 NVIDIA Driver 설치에 관한 내용을 다뤘습니다.

Ubuntu에 NVIDIA Driver 설치

 

 

이번 포스팅은 CUDA 설치 과정을 다뤄 보겠습니다.

 

CUDA를 설치하기 위해 NVIDIA Driver와의 버전 호환성을 확인해야 합니다.

아래 사이트에서 버전 호환성을 확인할 수 있습니다.

1. Why CUDA Compatibility — CUDA Compatibility r555 documentation

 

 

아래 표에서 NVIDIA Driver와 CUDA 버전 호환성을 확인할 수 있습니다.

저는 NVIDIA Driver 550 버전을 설치하였으므로 CUDA 12.4 버전을 설치하겠습니다.

NVIDIA Driver, CUDA 버전 호환성

 

 

호환성을 확인했으면 아래 사이트에서 CUDA를 다운 받으면 됩니다.

CUDA Toolkit Archive

 

 

설치할 버전을 클릭하고 아래와 같이 본인 PC 환경에 맞게 설정합니다.

저는 Linux > x86_64 > Ubuntu > 20.04 > deb (local) 순으로 설정하였습니다.

CUDA Toolkit Download

 

 

설정하면 아래에 명령어들이 나타나는데 순서대로 터미널에 입력하면 CUDA가 설치됩니다.

CUDA Toolkit Download

 

 

설치가 완료되면 CUDA가 설치된 경로를 시스템 경로에 추가하고,

CUDA 프로그램과 라이브러리가 시스템에서 원활히 동작할 수 있도록 설정해야 합니다.

 

아래 명령어를 통해 ~/.bashrc 파일을 엽니다.

~/.bashrc 파일은 사용자 환경 설정 파일로, 사용자가 터미널을 열 때마다 자동으로 실행되는 스크립트입니다.

gedit ~/.bashrc

 

 

아래 두 줄의 코드를 ~/.bashrc 가장 밑에 추가해 줍니다.

여기서 설치한 CUDA 버전에 맞게 코드를 수정하면 됩니다.

export PATH="/usr/local/cuda-12.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH"

 

 

수정을 하고 Ctrl + S로 저장하고, 아래 명령어를 통해 변경 사항을 적용합니다.

source ~/.bashrc

 

 

그러면 이제 아래 명령어를 통해 CUDA가 설치된 것을 확인할 수 있습니다.

nvcc -V

CUDA 버전 확인

 

 

다음 포스팅에서는 cuDNN 설치에 관한 내용을 소개하겠습니다.

Ubuntu에 cuDNN 설치