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