tensorflow gpu 사용하기

각종 블로그를 찾아봤지만 문제를 해결하지 못했는데, 과거 학부 교수님께서 올려주신 영상을 참고하여 문제를 해결했다.

환경: CUDA version: 12.0

1. Nvidia GPU 드라이버와 CUDA Toolkit을 설치해준다.

2. 아나콘다 환경에서 다음과 같이 입력한다. 꼭! 3.7 버전이어야 한다.
conda create -n tfgpu python=3.7
conda activate

3. 다음을 입력한다. 여기서도 꼭! 2.1 버전이어야 한다.
conda install tensorflow-gpu=2.1

설치 후, 각 패키지가 아래와 동일한 버전인지 확인한다. 버전이 조금이라도 다르면 GPU 인식이 안된다.
python                    3.7.13
cudatoolkit               10.1.243
cudnn                     7.6.5
tensorflow                2.1.0
tensorflow-base           2.1.0
tensorflow-estimator      2.1.0
tensorflow-gpu            2.1.0

아래와 같이 입력하고 로그가 동일하게 출력되는지 확인한다. 모든 줄에서 오류문구 없이 Successfully opened dynamic library가 출력되어야 한다. 한 줄이라도 에러 문구가 발생하면 GPU를 인식하지 못한다.

(tfgpu) C:\Users\User>python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
2023-02-25 19:21:23.112188: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2023-02-25 19:21:25.247015: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library nvcuda.dll
2023-02-25 19:21:25.274488: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1555] Found device 0 with properties:
pciBusID: 0000:01:00.0 name: NVIDIA GeForce GTX 1660 SUPER computeCapability: 7.5
coreClock: 1.83GHz coreCount: 22 deviceMemorySize: 6.00GiB deviceMemoryBandwidth: 312.97GiB/s
2023-02-25 19:21:25.274603: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2023-02-25 19:21:25.278710: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_10.dll
2023-02-25 19:21:25.282389: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_10.dll
2023-02-25 19:21:25.284107: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_10.dll
2023-02-25 19:21:25.287784: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_10.dll
2023-02-25 19:21:25.290019: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_10.dll
2023-02-25 19:21:25.297349: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
2023-02-25 19:21:25.297540: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1697] Adding visible gpu devices: 0
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]