Ubuntu 18.04에 NVIDIA CUDA 설치하기

이번에는 우분투에 nvidia CUDA를 설치해보록 할게요.

- 환경

1. Ubuntu 18.04 LTS

2. Nvidia GTX 660

- 필요 환경

1. 인터넷 환경

2. CUDA 설치를 위한 여유 공간

(설치 패키지 약 1.6GB, 설치 공간 약 5GB, CUDA 10.1 기준)

설치 패키지 다운로드를 위해 아래의 링크를 통해

NVIDIA CUDA 홈페이지에 접속하세요.

CUDA Toolkit Archive (CUDA Toolkit List)

그다음 사용할 CUDA 버전을 선택하세요.

저는 환경 요인 때문에 최신 버전이 아닌 CUDA 10.1 다운로드 했어요.

해당 CUDA 버전 다운로드 페이지로 이동하여 CUDA 설치할

OS, 아키텍쳐, 분류, 우분투 버전, 설치 타입 등을 선택하고

하단 오른쪽에 나오는 Download 버튼을 클릭하세요.

OS : Linux

Architecture : x86_64

Distribution : Ubuntu

Versuib : 사용 중인 Ubuntu 버전

Installer Type : deb - local

현재 사용 중인 Ubuntu 버전을 모르시는 분은 setting의

Details - About에서 Ubuntu 버전 정보를 볼 수 있어요.

저는 deb local로 다운로드해 터미널을 열어 deb 설치 파일이 있는

/Download 폴더로 이동하여 다음과 같은 명령어를 입력하여 설치를 시작했어요.

sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.105-418.39_1.0-1_amd64.deb

dpkg -i 명령이 끝나면 터미널 아래에 apt key를

추가하는 명령어가 보이는데 이를 복사하여 실행해주세요.

홈페이지에 써 있는 명령어의 <version> 란에 현재 설치하신

cuda-repo 경로를 넣어야 되니 주의하세요.

sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub

이제 우분투를 업데이트하고 CUDA를 설치해주세요.

CUDA 10.1 기준 4,933 MB의 저장 공간이 필요해요.

sudo apt-get update

sudo apt-get install cuda

다음으로는 환경 구축을 위해 ~/.profile에 CUDA 정보를 추가하세요.

1. 파일 열기

sudo gedit ~/.profile

2. ~/.profile에 정보 추가하고 저장하기

export PATH=/usr/local/cuda-10.1/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH

3. 변경 사항 적용하기

source ~/.profile

이제 CUDA 설치가 완료되었는데요.

CUDA 설치가 잘 되어 버전 정보가 잘 읽히나 확인해 볼게요.

터미널에 nvcc -V를 입력하여 아래에 설치한 버전 정보가 맞게 나오나 확인하세요.

또한 nvidia-smi을 통해서도 CUDA 버전을 확인할 수 있어요.

CUDA를 설치하면 그래픽 카드 드라이버도 같이

설치되는 것으로 알고 있어서 Ubuntu가 설치 초기 상태라면

CUDA만 바로 설치해도 될 듯하네요.

그럼 이상으로 포스팅을 마치며 오늘도 즐거운 하루 보내세요.

---

연관 포스팅

1. darknet YOLO로 물체 인식하기 (deep learning, object detection)