Linux

Rocky linux에서 Ndivia 드라이버 및 cuda 설치

콘파냐 2022. 4. 6. 18:54
반응형

CUDA 설치는 선택사항입니다.

CUDA는 GPU를 사용한 병렬처리, 예를들어 텐서플로우에서 병렬처리 사용 등이 아니라면 설치안해도 됩니다.

설치하기

  • EPEL 패키지 저장소 추가하기 - epel은 추가 패키지 저장소를 뜻한다.

     $ sudo dnf install epel-release
  • Ndivia driver 저장소를 추가하기

      $ sudo dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo
  • kernel devel과 header를 설치 - 그래픽 드라이버를 설치하기 위해서는 kernel devel과 header가 필요하기 때문에 설치합니다.

      $ sudo dnf install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
  • Nvidia 드라이버를 설치합니다.

      $ sudo dnf install nvidia-driver nvidia-settings
  • CUDA 드라이버를 설치합니다.(선택)

      $ sudo dnf install cuda-driver

    재실행합니다.

  • 설치 확인하기

      $ nvidia-smi

Ndivia 드라이버 삭제하기

$ sudo dnf remove nvidia-driver nvidia-settings cuda-driver kernel-devel-$(uname -r) kernel-headers-$(uname -r)  
반응형