본문 바로가기
공부/리눅스

CUDA, Docker 설치

by SLkittys 2024. 5. 3.
728x90

 

설치 전 사용하는 사용하는 OS 및 그래픽 카드 정보 확인

 

아래 링크에서 그래픽 드라이버 버전과 쿠다 버전의 상호 지원 여부 확인

CUDA Compatibility :: NVIDIA Data Center GPU Driver Documentation

 

 

그래픽 드라이버 확인 및 다운로드

공식 고급 드라이버 검색 | NVIDIA

 

CUDA 다운로드

CUDA Toolkit 12.2 Downloads | NVIDIA Developer

CUDA Toolkit 12.3 Update 2 Downloads | NVIDIA Developer

------------------------------------------------------------------

Cuda 설치

 

1. Nouveau 비활성화

 

nouveau 상태확인

$ lsmod |grep nouveau

nouveau 비활성화

vi /etc/modprobe.d/nouveau-blacklist.conf
blacklist nouveau
option nouveau modeset=0

초기 RAM 파일 시스템(initramfs)을 업데이트

update -initramfs -u

재시작

reboot

 

2. 그래픽 드라이버 설치

 

3. 쿠다설치

 

4. 완료 후 확인

nvcc --version

nvcc -v

 

* nvcc 안먹을때

  1. ~/.bashrc 수정(추가)

export PATH="/usr/local/<cuda버전/설치시 확인>/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/<cuda버전/설치시 확인>
/lib64:$LD_LIBRARY_PATH

 

  2. 터미널에 수정내용적용

source ~/.bashrc

ㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡ

Docker 설치

 

dnf 플러그인 설치

dnf -y install dnf-plugins-core

 

dnf config manager \

--Add-repo \

https://download.docker.com/linux/centos/docker-ce.repo

* rocky는 없음 centos 써도 무방

 

dnf install docker-ce docker-ce-cli containers.io

*오류 시

dnf install docker-ce docker-ce-cli containers.io \

 

--allowerasing

 

--skip-broken

--nobest

ㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡ

Nvidia-docker

 

기존 도커 지우기

dnf remove docker-ce

dnf install nvidia-docker2

 

도커 재시작

systemctl restart docker

 

설치확인

nvidia-docker

*관련 명령어 출력

 

참조

리눅스 플랫폼, 아키텍처, 시스템의 비트를 확인하는 방법 (scbyun.com)

NVIDIA CUDA Window에 설치하기(병렬처리, 엔비디아 쿠다 설치) (tistory.com)

 

728x90