Window Pycharm(파이참) GPU 연결 with tensorflow

먼저 https://kc9302.tistory.com/114 이걸 참고해서 1~4번까지 하고 나서 다음 내용을 확인해볼 수 있다.

구분	버전
GPU	GeForce GTX 1050 Ti
Anaconda	Anaconda3-2023.09-0-Windows-x86_64
Python	3.8
CUDA	cuda_11.2.0_460.89_win10
tensorflow-gpu	2.5.0
torchvision	0.9.0
torchaudio	0.8.0
cudatoolkit	11.1

이전에 설정한 환경과 동일하다. 하지만 처음 동작시 GPU를 인식하지 못했다.

원인은 tensorflow 버전이 맞지 않았다.

Tensorflow Check Config

https://spltech.co.uk/installing-tensorflow-2-9-with-gpu-support/ (이곳에서 버전에 맞는 환경 설정을 확인할 수 있다.)

CUDA 11.2를 설치했으니 Tensorflow 2.5.0을 설치한다. 여기서 Tensorflow를 설치하는 게 아닌 Tensorflow-GPU로 설치해준다.

conda install tensorflow-gpu==2.5.0

상황에 따라 pip로 설치해도 문제없다.

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

이후 이 코드를 실행했을 때 아래와 같이 GPU가 잡힌다면 올바르게 환경을 설정한 것이다.

Tensorflow GPU check