[Ubuntu20.04] Yolov5를 위한 우분투 세팅

졸업프로젝트로 Ubuntu20.04 환경의 서버를 발급받았는데 첫 학기, 방학, 두번째 학기 세번씩 새로 서버를 받아서 세팅하다보니 구글링하기도 귀찮아져서 정리해두기로 했다.

해야할 것:

• NVIDIA 드라이버 설치
• CUDA 설치
• cuDNN 설치(optional)
• Pytorch 설치

이 때, 미리 파이토치 지원 쿠다 버전을 확인해보자. 너무 최신 버전을 설치하면 Pytorch 호환이 안된다.

+ 할당받은 가상환경 확인

우분투 버전 확인 명령어

lsb_release -a

1. NVIDIA 드라이버 설치

(수정: 서버가 아닌 실물컴퓨터를 세팅할 경우 문제가 발생하여 다른 방식으로 세팅해주었다 https://grdnr13.tistory.com/107 참고)

우분투 NVIDIA Driver 설치 여부/버전 확인 명령어

cat /proc/driver/nidia/version

우분투 GPU 확인 명령어

nvidia-smi --query | fgrep 'Product Name'
# 또는 아래 명령어 사용
lspci | grep -i VGA

설치 가능한 드라이버 확인 명령어

ubuntu-drivers devices

나는 Command not found 에러가 떠서, 콘솔창에서 시키는 대로

sudo ubuntu-drivers autoinstall

위를 이용해 설치해줬다.

설치를 해주자

sudo apt update
sudo apt upgrade

sudo apt-get install nvidia-driver-525

그리고 reboot해준 뒤, 설치확인

sudo apt-get install nvidia-driver-470
# 아까랑 다르게 설치된 게 뜬다
nvidia-smi

왜 CUDA버전이 12.0으로 뜨는 건지 모르겠다

2. CUDA 설치

파이토치 사이트를 확인해보니 CUDA 11.8과 11.7이 지원됐다. (https://pytorch.org/get-started/locally/)

11.8을 설치해보도록 하겠다.

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html 참조

https://developer.nvidia.com/cuda-toolkit-archive 사이트에서 설치하려는 버전을 클릭한 뒤, 내가 사용중인 환경에 맞게 선택한다.

리눅스 아키텍쳐 (x86_64 등) 확인은 다음 명령어를 이용한다.

uname -m

그러면 아래에 명령어가 쭉 나온다. 하나씩 복붙&엔터로 실행시켜주자.

혹시모르니 CUDA 설치 확인 (apt install nvidia-cuda-toolkit으로 cuda toolkit 설치를 해줬다)

nvcc -V

3.Pytorch 설치

https://pytorch.org/ 에서 환경에 맞는 명령어를 받아온다.

cmd창에서 python3를 입력한 뒤, `import torch`를 해줬을 때 에러가 없으면 된다.