Django EC2에서 cuDNN 에러 해결하기

2020년 12월 07일 기록

Django EC2 환경에서 ai 모델 예측을 하려고 했더니 나온 에러

 Loaded runtime CuDNN library: 7.5.1 but source was compiled with: 7.6.4.  CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version. If using a binary install, upgrade your CuDNN library.  If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration.

7.6.4 버전이 필요한데 7.5.1 버전으로 컴파일 되었다고 한다.

$ nvcc --version

https://developer.nvidia.com/rdp/cudnn-archive 로 들어가서

맞는 버전 설치

EC2로 옮기고 (난 장고로 옮겼다)

$ tar xvfz 다운로드받은 cudnn파일명
$ which nvcc
-> /usr/local/cuda/bin/nvcc 이런식으로 폴더 경로 나올텐데 bin까지만 복사해주면 됨

$ sudo cp cuda/lib64/* /usr/local/cuda/lib64/
$ sudo cp cuda/include/* /usr/local/cuda/include/
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h

추가 dependencies 설치
$ sudo apt-get install libcupti-dev

환경변수 설정(맨 밑에 추가)
$vi ~/.bashrc

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64" export CUDA_HOME=/usr/local/cuda

cudnn 버전 확인
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

->
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 4
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"
대충 이런식으로 나오면 설치 끝!