Ubuntu 20.04 LTS (CLI-Server)에 CUDA 11.1 GPU 가속 연산 환경 세팅 하기

Ubuntu 20.04 LTS (CLI-Server)에 CUDA 11.1 GPU 가속 연산 환경 세팅 하기

1
AI 요약

이 글은 AI가 원문을 분석하여 핵심 내용을 요약한 것입니다.

Ubuntu 20.04 LTS CLI 서버에서 CUDA 11.1 GPU 가속 환경 세팅

이 게시물은 Ubuntu 20.04 LTS CLI(Server용) 환경에서 RTX 2080 Ti GPU 6대를 이용하여 CUDA 11.1 및 관련 Nvidia 드라이버, cuDNN 설치와 호환성 점검 방법을 상세히 설명합니다.

주요 설치 및 호환성 점검 내용

  • GPU 모델과 Nvidia 드라이버, CUDA 버전, AI 프레임워크(Pytorch, TensorFlow) 간 버전 호환성 확인
  • GUI 없이 CLI 서버에 최적화된 headless Nvidia 드라이버(nvidia-headless-450-server)와 CUDA 툴킷 설치 방법
  • CUDA 및 cuDNN 설치 후 시스템 변수 수동 설정 방법과 설치 확인 절차
  • 설치 완료 후 Pytorch 및 TensorFlow에서 GPU 가속 기능 확인하는 코드 예시 제공

설치 및 확인 참고사항

  • --no-install-recommends 옵션을 통해 GUI 컴포넌트 제외
  • nvidia-smi 명령어는 드라이버가 지원하는 CUDA 버전을 보여주며, 실제 설치된 CUDA 버전과 다를 수 있음