반응형
0. Overview
Nvidia의 인공지능 병렬처리를 위해서
다음이 필요하다.
- Nvidia Driver
- CUDA
- cuDNN
또한, 이들 간에 버전을 서로 맞춰주어야한다.
위 사진은
cuDNN 8.2.2와 호환되는 CUDA와 Driver의 버전이다.
아래의 링크에서 다른 cuDNN버전에 대해서도 확인가능하다.
https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html#fntarg_1
Support Matrix :: NVIDIA Deep Learning cuDNN Documentation
These support matrices provide a look into the supported versions of the OS, CUDA, the CUDA driver, and the NVIDIA hardware for the cuDNN 8.2.2 and earlier releases.
docs.nvidia.com
1. Install CUDA
CUDA는 아래의 링크에서 설치가능하다.
https://developer.nvidia.com/cuda-downloads
CUDA Toolkit 11.4 Update 1 Downloads
Resources CUDA Documentation/Release NotesMacOS Tools Training Sample Code Forums Archive of Previous CUDA Releases FAQ Open Source PackagesSubmit a Bug
developer.nvidia.com
Select env.
Example : CUDA 11.4.1
$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
$ sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-ubuntu1804-11-4-local/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get -y install cuda
반응형