Ubuntu 18.04, nvidia cuda 설치하기.

Ubuntu 18.04, nvidia cuda 설치

1. GPU 설치 확인
Tesla V100이 설치되어 있으며, device driver가 설치되어 있지 않아서 UNCLAIMED로 표시된다.

2. CUDA 설치(nvidia 디바이스 드라이버도 함께 설치)

다운로드 : https://developer.nvidia.com/cuda-downloads
다운로드 사이트에 들어가서 설치할 시스템에 맞는 버전을 선택한다. 아래는 ubuntu linux 64bit용이다.

다운로드 받은 폴더에서 다운로드 받은 파일을 실행한다.

만약, 컴파일러가 설치되어 있지 않아서 오류가 발생한다면, 아래 명령으로 컴파일러를 설치해 준다.

설치가 시작되고, 라이센스 동의 화면에서 yes 입력하면, 아래처럼 드라이버를 비롯해 cuda 툴킷을 설치할 수 있는 메뉴가 보인다.

install을 선택하고 enter 키를 누른다.
설치가 완료되면 아래와 같이 PATH, LD_LIBRARY_PATH 환경변수를 설정 방법과 삭제 방법을 안내해준다.

nvidia 드라이버 설치 확인

GPU 테스트(gpu_burn은 이전글 nvidia GPU 스트레스 테스트 참조)

gpu_burn으로 테스트하기위해 make하기전에 아래 환경변수 설정해야한다.

gpu 테스트까지 이상 없다면 CUDA를 사용할 준비가 끝났다.

3. nvidia 드라이버만 설치
여기서는 cuda를 사용하지 않고 nvidia device driver만 설치하는 방법이다.

드라이버 설치를 위한 repository를 등록한다.

위에서, 현재 드라이버 버전이 430인 것을 확인한다.

이제 아래 명령으로 드라이버를 설치하고 재부팅한다.

드라이버를 확인해 본다.

이제, UNCLAIMED 가 사라졌다.

nvidia-smi 명령으로 확인해 본다.

댓글 남기기

Your email address will not be published.

%d 블로거가 이것을 좋아합니다: