이제라도 기록하기

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

Settings/GPU

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

sssky00 2023. 8. 3. 18:13

더 빠른 실습 환경을 구축하기 위해 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

 

Wikiwand - CUDA

CUDA is a proprietary and closed source parallel computing platform and application programming interface that allows software to use certain types of graphics processing units for general purpose processing, an approach called general-purpose computing on

www.wikiwand.com

위 사이트에서 자신의 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

 

Windows의 소스에서 빌드  |  TensorFlow

이 페이지는 Cloud Translation API를 통해 번역되었습니다. Switch to English Windows의 소스에서 빌드 컬렉션을 사용해 정리하기 내 환경설정을 기준으로 콘텐츠를 저장하고 분류하세요. 소스에서 TensorFlow

www.tensorflow.org

 

최종 설치할 CUDA, cuDNN 버전
- CUDA 11.8
- cuDNN 8.6
저작자표시

'Settings > GPU' 카테고리의 다른 글

Window에서 PyTorch 설치 및 GPU 사용  (0) 2023.08.03
Window에서 Tensorflow 설치 및 GPU 사용  (0) 2023.08.03
(설치 후) GPU, CUDA, cuDNN 버전 확인  (0) 2023.08.03
Window에서 CUDA 11.8, cuDNN 8.7.0 설치  (0) 2023.08.03
'Settings/GPU' Related Articles more