[DATA/기타] Windows tensorflow-gpu 설치

CUDA, cuDNN 설치

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

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

위 링크에서 설치하려는 tensorflow-gpu 버전에 맞는 CUDA와 cuDNN을 확인하고 설치합니다.

(설치에 어려움이 있다면 여기를 참고하세요.)

tensorflow-gpu 설치

저는 버전 2.10.0을 사용했습니다. 아래 명령어를 입력하여 tensorflow-gpu를 설치해줍니다.

$ pip install tensorflow-gpu==2.10.0

tensorflow-gpu와 호환성이 좋은 numpy를 설치합니다. (제 경우에는 1.21이었습니다.)

$ pip install numpy==1.21
$ conda install numpy=1.21

설치 확인

import tensorflow as tf
from tensorflow.python.client import device_lib

print("TensorFlow version:", tf.__version__)
print(device_lib.list_local_devices())

위 코드를 실행했을 때 아래처럼 device에 GPU가 잡힌다면 정상적으로 설치 된 것입니다.