Ubuntu

Ubuntu 20.04 , CUDA 11.x , cuDNN 설치

YGspace 2021. 7. 16. 14:04

 Quick Guide

그래픽 드라이버 설치

sudo apt install nvidia-driver-460

CUDA 다운로드

wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run

CUDA 설치

sudo chmod +x cuda_11.1.1_455.32.00_linux.run
sudo sh cuda_11.1.1_455.32.00_linux.run

CUDA 경로 추가
nano ~/.bashrc

export PATH=$PATH:/usr/local/cuda-11.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
export CUDADIR=/usr/local/cuda-11.1
source ~/.bashrc
cuDNN 다운로드
https://developer.nvidia.com/cudnn

cuDNN 설치

tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

 

 

 

 

1. NVIDIA 드라이버 설치

sudo apt install nvidia-driver-460

 

2. CUDA 다운로드 (runfile local)

https://developer.nvidia.com/cuda-downloads?target_os=Linux 

 

CUDA Toolkit 11.4 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

2.1 버튼 선택 후 설치

Linux - x86_64 - Ubuntu - 20.04 - runfile(local)

 

터미널 명령 입력

 

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run

 

3. CUDA 설치

 

3.1 실행권한 추가

sudo chmod +x cuda_11.1.x_linux.run

 sudo chmod +x <CUDA파일명>

 

3.2 설치파일 실행

sudo sh cuda_11.4.0_470.42.01_linux.run

 

Continue


 

 

accept

이전에 NVIDIA 드라이버를 설치해줬기때문에, CUDA ToolKit만 설치해준다.

 

3.3 경로 추가

경로를 추가하기 위해 .bashrc 열기

nano ~/.bashrc

 

부분을 버전에 맞게 수정하여 맨 밑에 코드 3줄 추가하기 

export PATH=$PATH:/usr/local/cuda-11.x/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.x/lib64
export CUDADIR=/usr/local/cuda-11.x


ex) CUDA 11.2
export PATH=$PATH:/usr/local/cuda-11.2/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.2/lib64
export CUDADIR=/usr/local/cuda-11.2

 

다시 터미널로 나와서 bashrc 적용

source ~/.bashrc

 

 

 

4. CUDNN 다운로드

4.1 cuDNN 다운로드

https://developer.nvidia.com/cudnn

 

NVIDIA cuDNN

NVIDIA cuDNN The NVIDIA CUDA® Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. cuDNN provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, nor

developer.nvidia.com

로그인 후 다운로드 페이지 진입

설치한 CUDA와 맞는 버전이 나와있지 않다면, Archived cuDNN Releases로 이동한다.

 

 

설치한 CUDA와 맞는 버전을 찾아 다운로드 한다.
cuDNN Library for Linux(x86_68) 다운로드

 

 

 

 

 

4.2 cuDNN 설치

다시 이전 다운로드 페이지로 돌아가 설치 가이드로 이동한다.

 

Installation Guide 로 이동한다.

https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

 

Installation Guide :: NVIDIA Deep Learning cuDNN Documentation

The NVIDIA® CUDA® Deep Neural Network library™ (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. cuDNN provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, normalizati

docs.nvidia.com

 

 

가이드의 방법을 따라 설치한다.

 

tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
저작자표시