[버전관리] Ubuntu - Cuda - Cudnn

1. Ubuntu version check

$ lsb_release -a

No LSB modules are available.     
Distributor ID: Ubuntu            
Description:    Ubuntu 18.04.2 LTS
Release:        18.04             
Codename:       bionic            

2. CUDA version check

$ nvcc --version (혹은 nvcc -V)

$ locate cuda | grep /cuda$ (경로 확인)

3. Cudnn version check

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

4. Pytorch는 설치시 cudatoolkit을 제공한다. cudnn 까지 패키지로 묶어서 깔아준다.

$ conda install pytorch torchvision cudatoolkit=10.0 -c pytorch