[환경설정] Ubuntu에 cuDNN 설치하기

주로 Window 에서 CUDA와 cuDNN을 사용하다가 서버를 사용할 일이 생겼는데 CUDA는 설치되어있지만 cuDNN은 없다는 것을 알았다. 그래서 이번 글에서는 Ubuntu에 cuDNN을 설치해보도록 하겠다.

---

먼저 cuDNN을 웹에서부터 다운로드 해야한다. 이 과정이 총 시간의 70%이상을 차지한다..용량이 꽤 큰데 왜 그런지 필자는 네트워크가 엄청 느려서 소요시간이 3-4시간이었다...ㅠㅠ

 

cuDNN 설치는 

https://developer.nvidia.com/cudnn

NVIDIA cuDNN

 

이곳에서 Down Load cudnn을 클릭하면 된다. 

*단, 설치하려면 NVIDIA Login이 필요하니 회원가입을 해야한다. 

 

로그인을 하고나면 어떠한 동의를 하라 하고, agree를 클릭해주면 다음과 같이 주르륵 목록이 나온다.

 

 

필자는 처음에 Ubuntu라는 목록이 여러 개 있어서 잠시 혼란이 있었는데 다 무시해주면 된다.

필요한 파일은 이거 하나이다. 

이제 파일을 받았다면 cudnn-~.tgz라는 압축파일이 하나 있을 것이다.

 

tar -xvf cudnn~.tgz 명령어로 압축을 풀어준다.

aistudy@aistudy: ~$ tar -xvf cudnn-11.3-linux-x64-v8.2.1.32.tgz

그럼 cuda라는 폴더가 생성되고 그 안에 다음과 같은 폴더들이 있을 것이다.

이제 이 폴더들에 있는 파일들을 cuda 폴더에 옮겨주면 끝이다.

aistudy@aistudy:~$cp ./cuda/include/* /usr/local/cuda-11.1/include

aistudy@aistudy:~$cp ./cuda/lib64/* /usr/local/cuda-11.1/lib64

---

오늘은 cudnn설치를 해보았다. 별로 어려운 건 없지만 주의해야할 점은

cuda와 cudnn이 서로 버전이 맞아야하고, graphic driver나 여러 다른 문제가 생겨날 수도 있다...

 

처음 설치할 때 제대로 확인하고 설치하는 것을 권장한다.