윈도우(window) tensorflow GPU 인식불가 문제

윈도우, rtx 3090에서 tensorflow를 사용하는데 GPU를 인식하지 못한다.

 

우선 확인방법은 아래 코드를 돌려본다.

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices()) #GPU가 나오면 인식중인 것
print(tf.test.is_built_with_cuda()) #True가 나오면 cuda 있음

첫번째 print에서 GPU가 나오면 일단 tf가 GPU를 잡은 것이라 아마 이 글은 도움이 되지 않을것이다.

나의 경우는 두번째 print에서도 False가 나왔다.

그래서 버전문제인가 의심했고 결국 버전을 다시 깔기에 이른다..

 

 

결론만 말하자면,

tensorflow 일정버전 이상에서는 tensorflow-gpu를 지원하지않고,

따라서 윈도우에서는 GPU+tf 조합을 쓸 수 없다..

tensorflow를 다운그레이드 해주어야한다.

이때 tensorflow-gpu 호환성을 보고 tensorflow 버전을 골라주자.

 

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

Windows의 소스에서 빌드  |  TensorFlow

 

여기서 cuda, cudnn에 맞게 tensorflow를 다시 깔아야한다.

그리고나서 tf에 맞게 CUDA, cudnn도 다시 깔아야한다

참, 이미 CUDA를 깐 사람을 위해 귀찮게 다 지우지않고 CUDA만 다시 까는 방법이 있다.

아래 링크 참고!

 

https://tw0226.tistory.com/79

[CUDA] 윈도우 환경에서 CUDA 버전 변경