[Linux] Ubuntu에서 CUDA 설치하기

테스트 환경

NCP 서버 1대 생성

OS : Ubuntu 20.04

 

1) 기존에 설치되어있는 cuda 삭제

apt-get purge nvidia*
apt-get autoremove
apt-get autoclean
rm -rf /usr/local/cuda*

 

2) cuda 다운로드 페이지에서 원하는 버전 선택

다운로드 페이지 링크 : https://developer.nvidia.com/cuda-toolkit-archive

OS 정보와 인스톨 타입을 선택하면 아래에 인스톨 명령어 제공, 해당 명령어로 설치 진행

wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run sh cuda_11.7.0_515.43.04_linux.run

 

설치화면으로 넘어가면 accept를 치고 설치 진행

드라이버 선택 후, 아래에 Install 선택

 

 

3) 버전정보 확인을 위한 환경변수 설정파일 세팅

방법1) profile 파일 직접 수정

vi /etc/profile

[아래 내용 맨 아랫부분에 추가]

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

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

export CUDARDIR=/usr/local/cuda-11.7

 

source /etc/profile

 

방법2) 명령어로 파일에 내용 수정

sh -c "echo 'export PATH=$PATH:/usr/local/cuda-11.7/bin'>> /etc/profile"

sh -c "echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.7/lib64'>> /etc/profile"

sh -c "echo 'export CUDARDIR=/usr/local/cuda-11.7'>> /etc/profile"

 

source /etc/profile

 

※ 방법1,2 의 버전이 적힌 부분(11.7)은 설치한 cuda의 버전정보 입력

 

4) 버전정보 조회로 설치 확인

nvcc -V