[Tensorflow] Tensorflow와 CuDNN 버전 에러

반응형

- 전체에러 문구

E tensorflow/stream_executor/cuda/cuda_dnn.cc:378] Loaded runtime CuDNN library: 7603 (compatibility version 7600) but source was compiled with 7004 (compatibility version 7000). If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.

F tensorflow/core/kernels/conv_ops.cc:717] Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms)

​

​

Tensorflow와 현재 CuDNN 사이의 버전이 맞지 않는 경우 발생하는 에러입니다. 저의 경우 각각의 사용버전은 아래와 같습니다.

사용 Tensorflow 버전: Tensorflow-gpu 1.5.0

사용 CUDA 버전: 9.0

​

에러 문구의 가장 앞쪽에 표시된 7603( CuDNN 7.6.3버전에 해당)으로 새로 설치하여 버전을 수정함으로써 해결할 수 있습니다!

​

- CuDNN 설치 방법

1. 아래 사이트에서 사용 CUDA버전과 맞는 원하는 CuDNN 버전을 다운로드해줍니다.

https://developer.nvidia.com/cudnn

CUDA Deep Neural Network

 

2. 다운로드된 폴더로 이동하고 아래 명령어로 압축을 풀어줍니다.

tar xzvf "파일 이름"
예시) tar xzvf cudnn-9.0-linux-x64-v7.6.3.30.solitairetheme8

3. 아래 명령어를 이용해 기타 패키지들을 설치해줍니다.

sudo apt-get install libcupti-dev

4. 아래 명령어를 이용해 맞는 버전이 설치되었는지, 설치된 버전을 확인해 줍니다.

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2