tensorflow gpu 사용 호환 및 버전 확인, 셋팅 방법

1. 내 gpu 확인하기 

 

시작메뉴에서 장치관리자 접근 -> 내 gpu 확인하기 

 

 

 

2. cuda에서 지원하는 gpu 버전 확인,

Compute Capability (CUDA SDK support vs. Microarchitecture)CUDA SDKVersion 보기

 

아래 이미지와 같이 gpu  명을 확인하고 가장 좌측의 capability 버전을 기억하면된다.

https://www.wikiwand.com/en/CUDA#/GPUs_supported

Wikiwand - CUDA

 

capability에 맞는 버전(좌측열) 확인

 

 

3. cudnn 및 cuda 버전 확인 (호환성)

 

아래의 링크에서 텐서플로우에서 지원하는 파이썬,

cuDNN, 쿠다 버전을 확인한다.

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

Windows의 소스에서 빌드,Windows의 소스에서 빌드  |  TensorFlow

 

그 다음에 각각 버전에 맞는 s/w 설치를 한다.

 

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

cuDNN Archive

 

 

 

https://developer.nvidia.com/cuda-11.2.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

CUDA Toolkit 11.2 Update 2 Downloads

 

 

마지막으로 아래 링크를 참고하여 

cuda 폴더에 cudnn 라이브러리 등을 넣어주면 끝!

 

https://sodayeong.tistory.com/227