(설치 전) GPU, CUDA, cuDNN 버전 확인

더 빠른 실습 환경을 구축하기 위해 CUDA, cuDNN를 설치하여 GPU 환경을 설정할 수 있다. 본인이 여러 사이트를 찾아보며 설치하고 실행했던 과정을 정리해보았다.

기준환경

• Window 11
• NIVIDA GeForce RTX 4070 Ti

1. CUDA Compute Capability

cuda 설치 전 cuda compute capability 확인하기
https://www.wikiwand.com/en/CUDA#/GPUs_supported

위 사이트에서 자신의 GPU에 해당하는 Compute capability를 찾고 Compute capability에 맞는 CUDA 버전을 찾을 수 있다.

2. 설치할 CUDA, cuDNN 버전 확인

tensorflow-gpu를 사용할 예정이라면 자신이 사용하는 tensorflow gpu와 호환되는 CUDA, cuDNN 버전을 아래 링크에서 확인한다.

https://www.tensorflow.org/install/source_windows?hl=ko#tested_build_configurations

최종 설치할 CUDA, cuDNN 버전

• CUDA 11.8
• cuDNN 8.6