[Ubuntu] Nvidia, CUDA 버전 확인하기

# 학부연구생 활동을 하며 Nvidia Tao-Toolkit을 사용하기 위해 Ubuntu 20.04 환경을 세팅해야 했다..!!

# 몇 주간의 미친듯한 구글링 결과를 남기기 위해 기록을 작성한다..

---

우분투 버전 확인

lsb_release -a

 

Nvidia 시스템 관리 인터페이스 확인

nvidia-smi

 

CUDA 버전 확인

nvcc -V

nvidia-smi를 입력했을 때에도 CUDA Version이 나오지만, 이는 깔려있는 Nvidia Drivier Version에 맞는 Version을 추천해주는 것이지 내 컴퓨터에 해당 CUDA Version이 깔려져 있다는 뜻이 아니다!! (깔려있는 줄 알고 고생했다..)

 

cudnn 버전 확인

# cudnn 8.x 이전
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

# cudnn 8.x 이상
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

Nvidia-container-toolkit 설치 확인

nvidia-container-cli info

 

Nvidia-container-runtime 설치 확인

which nvidia-container-runtime-hook

 

Nvidia-docker2 설치 확인

nvidia-docker version