NVIDIA cuDNN(CUDA Deep Neural Network)은 심층 신경망에서 사용하기 위한 GPU 가속 프리미티브 라이브러리입니다.

사전 요구 사항

  • NVIDIA CUDA 버전과 일치하고 Linux 배포에 적합한 cuDNN 패키지를 다운로드할 NVIDIA 개발자 계정을 생성합니다. https://developer.nvidia.com/cudnn을 참조하십시오.
  • vSphere Bitfusion 클라이언트를 설치했는지 확인합니다.
  • NVIDIA CUDA를 설치했는지 확인합니다.

프로시저

  1. Linux 배포에 대한 명령 시퀀스를 실행하여 NVIDIA cuDNN 패키지를 설치합니다.
    • Ubuntu 버전 20.04
      sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
    • CentOS 8 및 Red Hat Linux 8
      sudo rpm -ivh libcudnn8-8.0.5.39-1.cuda11.0.x86_64.rpm
  2. (선택 사항) NVIDIA cuDNN이 설치되었는지 확인하려면 ldconfig -p | grep cudnn을 실행합니다.

다음에 수행할 작업

  • CentOS 또는 Red Hat Linux를 사용하는 경우에는 먼저 Python 3을 설치해야 합니다. CentOS 및 Red Hat Linux에 Python을 설치하는 방법의 내용을 참조하십시오.
  • Ubuntu Linux를 사용하는 경우 TensorFlow, PyTorch 및 YOLO를 설치할 수 있습니다.