※ GPU를 사용하기 위한 cuda 및 GPU 드라이버 설치
※ 사용환경 : Ubuntu LTS 20, RTX 3090
0. 사용환경 update
$ sudo apt update
Error : sudo command not found 가 뜬다면
$ apt-get install -y sudo
1. nvidia-driver 설치
$ sudo apt search nvidia-driver
해당 apt 버전에 맞는 nvidia 드라이버 버전을 보여준다. 나오는 드라이버 중 원하는 것을 설치하면 되는데, 최신 버전은 아직 불안정할 수 있으므로 최신버전의 아래 단계를 추천한다.
아니면
$ ubuntu-drivers devices
위의 명령어로 우분투에서 추천하는 드라이버를 확인하는 방법도 있다.
$ sudo apt-get install nvidia-driver-460 (460 대신 원하는 버전의 숫자를 입력하면 된다. 서버버전 아님! )
2. CUDA 설치
$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
$ sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
$ sudo apt-get update
$ sudo apt-get -y install cuda-11-1
$ sudo apt install nvidia-cuda-toolkit
3. 설치 확인
$ reboot 또는 컴퓨터 다시 시작
컴퓨터를 재시작해야 해당 위의 작업들이 컴퓨터에 반영되니 꼭 다시 시작하기!
$ nvcc -V (cuda-toolkit 설치 확인)
$ nvidia-smi (nvidia-driver 설치 확인)