When will you grow up?

[Error]Could not find 'cudnn64_6.dll' 본문

02. Study/Tensorflow

[Error]Could not find 'cudnn64_6.dll'

미카이 2018. 1. 7. 20:15

Tensorflow 1.2.1 version을 사용하는 도중에 버전을 업데이트 해야할것 같아서,

1.4.0 버전으로 업데이트를 진행하였다.


이전버전을 지우고

pip uninstall tensorflow-gpu 로 지운다음


1.4.0버전을 다운받았다

pip install https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.4.0-cp35-cp35m-win_amd64.whl


그런데 갑자기 아래와 같은 오류가 나왔다


Error

Could not find 'cudnn64_6.dll'


이건 뭥미?? 이 오류는 무엇??


구글링을 하여 간단하게 해결을 하였다.


1.2버전까지는 cuDNN5.1 을 사용하였는데 1.3이상 부터는 cuDNN6을 사용해야 된다는것!!


그래서 NVIDIA에 들어가서 cuDNN6을 찾아보았는데


https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v6/prod/8.0_20170307/cudnn-8.0-windows10-x64-v6.0-zip

로 들어가서 받을수 있다.


그리고 나서 압축풀고 bin, include, lib 파일 복사한후

"C : \ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v8.0 안에다가 붙여넣기하고 다시 학습을 진행하면 잘 해결된것을 확인할 수 있다.


pre-training된 모델도 사용해보고 이것저것 해봐야겠다.

Comments