우분투 18.04에 NVIDIA DRIVER 재설치 (+cudnn 8.2.0, cuda 11.2 버전 확인)

Nvidia driver 재설치

0. nvidia-smi로 driver 잘 설치되었나 확인

 

1. 기존 nvidia driver 삭제

sudo apt-get --purge remove *nvidia*

2. 내 버전에 알맞는 nvidia driver 확인

ubuntu-drivers devices

 

3. 호환되는 버전의 nvidia driver 설치

sudo apt install nvidia-driver-470

 

4. sudo reboot

 

 

cuda 버전 확인

nvcc -V

 

cudnn 버전 확인 (8버전 이상)

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

 

(8버전 이하 : cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2)

 

토치 1.10.1 재설치

https://pytorch.org/get-started/previous-versions/

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html