cuda, cudnn, nvidia driver 버전 확인 + cudnn 설치/제거

cuda, cudnn 버전 확인

• 순서대로 cuda, cudnn 버전

• cudnn은 버전에 따라 파일이 다름

nvcc -V  

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

nvidia driver 버전 확인

apt --installed list | grep nvidia-driver

pytorch로 확인하기

import torch

print("cudnn version:{}".format(torch.backends.cudnn.version()))
print("cuda version: {}".format(torch.version.cuda))

cudnn 설치

https://developer.nvidia.com/rdp/cudnn-archive

• 본인이 설치하고 싶은 버전으로 설치하면 됨.

  • for Linux x86_64 (Tar) 우분투여도 이것으로 설치하기

# 본인이 다운받은 압축파일을 풀면 됨
$ tar -xvf cudnn-linux-x86_64-8.9.3.28_cuda11-archive.tar.xz

# 파일을 cuda에 복사하기
$ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
$ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

cudnn 제거

• cuda가 저장된 경로의 lib64 를 찾아서 libcudnn으로 시작하는 파일 지우기

sudo rm -r /usr/local/cuda-11.6/lib64/libcudnn*