Anaconda tensorflow-gpu 설치 에러 [WinError 126] 지정된 모듈을 찾을 수 없습니다, importError: Could not find 'cudnn64_6.dll'
anaconda를 이용하여 tensorflow cpu 버전을 사용하다가 training이 끝나지를 않아서
gpu 버전을 써보기로 했다.
cpu 버전은 나름 쉽게 설치했었는데 gpu는 조금 더 까다로웠다.
블로그들을 참고해보니 막 무슨 파일을 다운 받고 어디에 붙여넣고 하라는데
처음에는 이해가 잘 안됐다.
내가 이해한대로 해보자면,
우선 다른 블로그를 참고해서 아나콘다와 tensorflow cpu 버전 설치 직전까지 진행해주시고,
필요한 파일은
CUDA 8.0
CuDNN 6.0
두 파일을 순서대로 설치.
대충 내용을 보니
CUDA 8.0은 그냥 설치하는 파일이고 CuDNN는 설치한 cuda 폴더 안에 추가하는
라이브러리, 기타등등 파일인데 다른 블로그에서는 CuDNN 폴더를 다 붙여 넣으라고도
했지만 딱 봐도 확연히 안에 들어있는 파일이 달라서 CuDNN 폴더에 있는 파일을
기존 cuda 폴더 안에 붙여 넣어주었다.
아무튼 여기서 주의해야 될 점은 CuDNN이 6.0이라는 것이다.
다른 블로그에서는 5.1을 받으라고 되어있는데
이상하게 내 anaconda에서는 계속 cudnn64_6.dll 파일을 찾을 수 없다고 에러가 떴다.
바로 이 에러이다.
알아보니 5.1은 bin 폴더안에 cudnn64_5.dll를 가지고 있고 6.0이 cudnn64_6.dll을 갖고 있었다.
cudnn64_6.dll을 기본 설치 폴더 기준
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin 안에 붙여넣으니
성공적으로 tensorflow-gpu를 이용할 수 있게 되었다.
0 comments