컴퓨터 비전 - 초기 환경 설정

OS	Ubuntu 20.04
GPU	NVIDIA GP104 [GeForce GTX 1070]
Architecture	x86_64

 

NVIDIA 드라이버 설치

nouveau를 삭제를 먼저 할 경우 재부팅이 되지 않는 등의 에러를 계속 일으켜 드라이버 설치 후 존재 여부를 확인하기도 함.

repository 생성 후 NVIDIA 드라이버를 설치할 경우 nouveau는 알아서 제거된다고 한다.

1. 운영체제 업데이트

sudo apt update && sudo apt upgrade -y

2. 그래픽 카드 정보 확인

ubuntu-drivers devices

권장 버전 확인

3. ubuntu-drivers 다운로드

# 권장버전 다운로드 시 sudo ubuntu-drivers autoinstall # 버전을 선택적으로 다운로드 시 sudo apt-get install nvidia-drvier-460

4. 재부팅

reboot

5. 드라이버 설치 확인 및 버전 확인

nvidia-smi

Nvidia 드라이버 설치 여부와 함께 권장되는 CUDA의 버전을 확인할 수 있음

6. nouveau 확인

lsmod | grep nouveau

존재 시 제거할 것

 

CUDA 설치 - 11.2

GPU 드라이버와 CUDA 버전 호환성을 확인하였고, 작성자가 설치한 드라이버는 460.91.03으로, 11.2를 설치하면 되는 것을 확인함.

호환성 확인: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-major-component-versions__table-cuda-toolkit-driver-versions

https://developer.nvidia.com/cuda-toolkit-archive 에 접속해서 원하는 버전의 쿠다를 선택하면 사용자의 환경에 맞는 쿠다 설치 명령어 등을 제공함.

선택을 완료하면 밑에 선택한 항목에 맞게 명령어가 뜬다.

(본 명령어는 작성자 기준)

wget https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux.run

CUDA 설치는 4-5시간 정도 걸린다.

이후

sudo sh cuda_11.2.2_460.32.03_linux.run

설치를 완료한 후 .bashrc에 아래 내용을 붙여 넣는다

vim .bashrc

# >>> cuda 11.2 >>>  export PATH=/usr/local/cuda-11.2/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}  # << cuda 11.2 <<

.bashrc를 실행시키고 설치가 잘 되었는지 확인한다.

설치가 잘 되었으면 아래와 같이 뜬다

source ~/.bashrc nvcc --version

root@Ubuntu:~# nvcc --version nvcc:
NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

 

cuDNN 설치 - 8.1.1

https://developer.nvidia.com/cudnn에 접속해서 Download cuDNN을 누르고 회원가입 한다.

설치한 CUDA와 호환되는 버전을 다운로드 한다.

11.2의 리눅스 cuDNN 라이브러리를 다운 받았다.

설치된 디렉토리로 이동해 압축을 해제한다.

tar –xzvf cudnn-11.2-linux-x64-v8.1.1.33.tgz

설치

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include sudo cp cuda/lib64/libcudnn.* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

설치 확인

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn

설치된 버전들

Nvidia-driver	460
CUDA	11.2.2
cuDNN	8.1.1