[Error]Could not find 'cudnn64

02. Study/Tensorflow

미카이 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을 찾아보았는데

로 들어가서 받을수 있다.

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

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

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