Ubuntu 18.04에 cuDNN 7.6.5 설치

Speedwell🍀·2022년 2월 16일
0

Human Following Robot

목록 보기
11/18

YOLOv4 공식 깃허브에서 CuDNN 버전은 8.0.2 이상이 좋다고 한다.
하지만 tensorflow에 맞춰서 7.6.5를 설치하려고 한다.

먼저 아래의 사이트에 접속한다.
https://developer.nvidia.com/rdp/cudnn-archive

이 중 나에게 맞는 cuDNN v7.6.5 for CUDA 10.2를 선택한다.

여기서 cuDNN Runtime Library for Ubuntu 18.04 (Deb), cuDNN Developer Library for Ubuntu 18.04 (Deb), cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)를 다운받는다.

아래의 명령어를 통해 cuDNN을 설치하면 된다.
$ sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb

그런데 이때 의존성 문제로 오류가 생겼다.

$ sudo dpkg -i libcudnn7*.deb을 해주니 해결되었다.

$ cat /usr/include/cudnn.h | grep -E "CUDNN_MAJOR|CUDNN_MINOR|CUDNN_PATCHLEVEL"로 cuDNN 버전을 출력했더니 잘 나왔다.

0개의 댓글