[Python] Tensorflow cudart64 not found error 해결

Python에서 tensorflow를 설치한 뒤 import 하면 아래와 같이 에러가 발생한다.

2021-09-01 10:47:43.744790: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
2021-09-01 10:47:43.745301: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
2021-09-01 10:47:47.810894: E tensorflow/core/lib/monitoring/collection_registry.cc:77] Cannot register 2 metrics with the same name: /tensorflow/api/keras/dropout/temp_rate_is_zero

 

아래의 링크로 접속하여 CUDA Toolkit을 다운로드 한 뒤 설치한다.

 

https://developer.nvidia.com/cuda-11.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork 

CUDA Toolkit 11.0 Download

 

설치시간은 금방되진 않는다.

 

설치한 뒤 다시 실행하면 에러가 뜰 수도 있다.

 

이전 경우는 설치도 안되어있고 설치가 안되어있으니 당연히 dll 파일이 없는 것이고

 

설치를 한경우는 진짜로 dll 파일이 있으나 못찾는 경우이다.

 

재부팅 후에도 못찾는다면 시스템 환경변수(또는 유저 환경변수) 에 dll파일이 있는 위치를 추가하자.

 

나의 컴퓨터 기준 cudart64_110.dll의 위치는 

 

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin 이다

 

 

참고로 tensorflow 버전에 따라 설치해야할 cuda 버전이 달라질 수 있으니 참고하자