Window 10
CUDA 10.1 환경에서,
pip install tensorflow 로 설치 시, Could not load dynamic library 'cudnn64_110.dll', dlerror: cudnn64_110.dll not found 에러가 발생하여, GPU가 잡히지 않는 에러가 발생한 경우 해결방법입니다.
해답을 먼저 말하자면,
CUDA 10.1 환경에서 pip로 tensorflow를 설치할 경우, GPU를 사용할 수 없습니다.
그 이유는 현재(20.12.15 이후) pip install tensorflow를 입력할 경우, tensorflow==2.4.0이 설치되기 때문입니다.
tensorflow==2.4.0은 아래 링크에서 볼 수 있듯이, CUDA 11.0을 기준으로 구성되었기 때문입니다. 또한 cuDNN 8.0을 바탕으로 구성되어 있습니다.
https://www.tensorflow.org/install/source_windows?hl=ko#gpu
Windows의 소스에서 빌드 | TensorFlow
소스에서 TensorFlow pip 패키지를 빌드하고 Windows에 설치합니다. 참고: 잘 테스트되고 사전 빌드된 Windows 시스템용 TensorFlow 패키지가 이미 제공되고 있습니다. Windows용 설정 다음 빌드 도구를 설치
www.tensorflow.org
따라서
1. CUDA 10.1으로 tensorflow GPU를 사용하셨다면, pip install tensorflow==2.3.0 을 통해, GPU를 사용하시거나,
2. CUDA 11.0 및 cuDNN 8.0 버전으로 업데이트하시고, tensorflow==2.4.0 버전의 GPU를 사용하시면 됩니다.
'Tensorflow' 카테고리의 다른 글
| triplet loss를 활용한 이미지 유사도 측정 (1) | 2021.10.07 |
|---|---|
| 협업 필터링을 활용한 영화 추천 (0) | 2020.12.19 |
| IMDB에 Bidirectional LSTM 모델 적용하기 (0) | 2020.12.17 |
| 케라스: 시계열을 활용한 기상(weather) 예측 (2) | 2020.12.16 |
| Tensorflow 콜백함수: ReduceLROnPlateau (2) | 2020.11.18 |