Tensorflow Pytorch GPU 설치를 위해 Nvidia CUDA cuDNN 버젼 확인하는 방법

Tensorflow에서 지원하는 버젼이 있기 때문에 그 이상의 최신 버젼을 설치하였을 때에는 오히려 동작하지 못하게 된다. 아래의 순서에 따라 설치하기 바란다. Pytorch는 Tensorflow와 지원하는 버젼이 다르니 Pytorch는 최신 지원하는 버젼을 확인하고 설치는 아래의 과정을 참조하면 된다. 즉, 현재 각각 지원하는 버젼은 Tensorflow는 CUDA 11.2 이고 Pytorch는 CUDA 11.8 이다.

Pytorch CUDA 버젼

 

1. Tensorflow 권장 사양 확인

https://www.tensorflow.org/install/gpu?hl=ko 

 

2. GPU 연산능력 확인

텐서플로우에서는 CUDA® 아키텍처 3.5, 5.0, 6.0, 7.0, 7.5, 8.0 이상을 포함하는 NVIDIA® GPU 카드만을 지원하기 때문에 Compute Capability를 확인하여야 한다.

https://developer.nvidia.com/cuda-gpus

 

3. CUDA 버젼

CUDA 11.2 버젼을 CUDA Toolkit ARchive 웹페이지에서 찾아서 설치한다.

https://developer.nvidia.com/cuda-toolkit-archive

 

4. cuDNN 버젼

cuDNN 버젼이 8.1이면서 CUDA 11.2를 지원하는 버젼을 찾아서 설치한다.

 

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