[DL][Windows] 윈도우 텐서플로우 (TensorFlow) GPU 인식 문제

 

0. 증상

 

TensorFlow 설치 - 정상

NVIDIA 그래픽 드라이버 설치 - 정상

CUDA 설치 - 정상
CUDNN 설치 - 정상

 

인식 확인 방법들 아래 참고

[확인한 리스트들]

텐서플로우 설치 확인 & 버전 확인

후술하겠지만, 텐서플로우 버전 문제였음..

 

CUDA 확인

CUDNN은 CUDA 디렉토리 확인 & 환경변수 확인

NVIDIA 그래픽 드라이버 확인

 

 

위 처럼 필요한 것들이 모두 설치 되었음에도 TensorFlow가 gpu를 인식하지 못한다면 버전 문제를 의심해 봅시다.

 

---

 

1. 원인 - TensorFlow 버전

 

윈도우 사용자 분들 환정으로 작성된 글입니다.

많은 분들이 TensorFolw 설치 시, 아래 명령어 들 중 하나를 선택해서 설치 하셨을 겁니다.

 

conda install tensorflow #1. conda 설치 시

pip install tensorflow #2. pip 설치 시

 

두 명령어 모두 tensorflow 버전을 지정하지 않았는데, 저장소 기준으로 최신 버전을 받아오게 됩니다.

이때, 문제점은 TensorFlow의 윈도우 네이티브 지원 버전 중 GPU를 지원하는 버전은 2.10이 마지막입니다.

 

 

https://www.tensorflow.org/install/pip?hl=ko#windows-native

 

---

 

2. 해결방법

 

설치된 tensorflow를 삭제하고 2.10 버전으로 재설치하는 방법으로 해결 가능합니다.

#conda 사용 시
conda uninstall tensorflow

conda list #tensorflow 관련된 패키지 확인

conda install tensorflow==2.10


#pip 사용 시
pip uninstall tensorflow

pip list #tensorflow 관련된 패키지 확인

pip install "tensorflow<2.11"


#GPU 설정 확인
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

#명령어 입력 시, GPU 장치가 반환되면 설치 완료입니다.

 

 

tensorflow와 함께 설치되는 패키지

 

위 이미지 처럼 tensorflow와 함께 설치되는 패키지들도 삭제 한 뒤, 재설치 해주셔야합니다.

 

---

 

3. 번외

conda 혹은 python venv 등 윈도우 기반 가상환경 사용 외에도

윈도우에서 기본적으로 제공하는 WSL2를 사용하여 환경 구성을 하시는 것도 하나의 방법입니다.