[Tensorflow] tensorflow-gpu 실행 오류

보다 더 빠른 학습을 위해서 tensorflow GPU 버전을 사용하려고 tensorflow와 tensorflow-gpu 패키지를 설치했는데, 사용하는 과정에서 아래와 같은 오류가 발생했다. 

Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found

오류가 났을 때 버전은 아래와 같다. 

tensorflow: 2.5.0 / CUDA: 10.0 / cuDNN: 7.4.5

 

python shell통해 확인

위의 사진을 보면 내가 지금 GPU를 사용하고 있는지 코드 상으로 확인해본 건데, tensorflow-gpu를 깔았음에도 불구하고 오류메세지와 CPU 표시만 있고 GPU가 잡혀있지 않은 것을 확인할 수 있다.  

GPU를 쓰는지 확인할 수 있는 코드는 아래와 같다. 

import tensorflow as tf
tf.__version__                    # 텐서플로 버전확인
from tensorflow.python.client import device_lib
device_lib.list_local_devices()   # GPU를 사용하는 지 확인

 

---

 

원인

tensorflow와 CUDA 버전이 맞지않아서 발생.

위의 오류메세지를 자세히 보면 'cudart64_110.dll' 를 로드할 수 없다고 하는데 이 파일은 cuda 11을 설치했을 때 존재하는 파일이다. 나는 CUDA 10버전을 사용하고 있으니 있을리 없다. 

 

해결

현재 CUDA 버전에 맞춰 tensorflow 버전을 변경하던지, tensorflow 버전에 맞게 CUDA를 변경하면 된다. 

나는 귀찮아서 전자를 택했다. 

아래 링크를 참고하면 tensorflow에 맞는 CUDA 버전을 안내해준다. 

https://www.tensorflow.org/install/source_windows#gpu

Windows의 소스에서 빌드  |  TensorFlow

버전을 맞추고 GPU가 잡힌 모습