[우분투] gpu사용을 위한, Nvidia cuda, cuda Dnn 설치하기 (feat.우분투)

환경 셋팅 

 

우분투 18.04 환경 

그래픽카드 버전 : 470.129.06

cuda : 11.4 

 

 

 

**gpu를 사용하기 위해 쿠다와 쿠다DNN을 설치해야 하는데 그전에 먼저 그래픽카드 드라이버가가 설치되어 있어야한다. **

 

# CUDA 삭제 sudo apt remove --purge "*cublas*" "cuda*" "nsight*"

# CUDA 파일 삭제 /usr/local/cuda-9.0/bin/uninstall_cuda_toolkit_9.0.pl rm -rf /usr/local/cuda-10.2/

 

-----------------------------<설치순서>------------------------------

Nvidia 그래픽 카드 드라이버 -> cuda toolkit -> cudaDNN

 

1. 그래픽 드라이버를 설치한다.

 

2. cuda toolkit 사이트 들어가서 해당 cuda 버전을 다운 받는다. 

 

cuda toolkit 사이트

 

https://developer.nvidia.com/cuda-toolkit-archive

CUDA Toolkit Archive

 

3. 설치 후 터미널에 입력해준다. 

- 노란 부분을 본인이 설치한 cuda버전에 맞게 수정해준다. 

 

export PATH=$PATH:/usr/local/cuda-11.4/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.4/lib64
export CUDADIR=/usr/local/cuda-11.4

 

 

 

3. source /etc/profile 실행 

 

4. cuda DNN사이트에서 cuda버전에 맞게 다운 받는다. 

 

https://developer.nvidia.com/cudnn

CUDA Deep Neural Network

 

4. tar 압축 해제 후 폴더 옮겨주기 

 

다운로드한 tar파일을 압축풀기한다. 

tar -xvzf 압축명.tar 

 

압축후 쿠다 폴더가 생성되는데 터미널에 아래와 같이 입력해준다. 이 때 연두색과 분홍색 칠해진 곳을 참고하여

경로를 입력해준다. 

 

cp 압축푼 쿠다폴더/파일 옮길 쿠다 폴더/파일  

 

sudo cp cuda/include/cudnn* /usr/local/cuda-11.4/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.4/lib64
sudo chmod a+r /usr/local/cuda-11.4/include/cudnn.h /usr/local/cuda-11.4/lib64/libcudnn*

 

 

*설치확인 

{그래픽 드라이버 설치 확인 : nvidia-smi,

  쿠다 설치 확인  : nvcc --version and nvcc -V

 쿠다DNN 확인 : cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2}

 

 

cuda DNN 정상적으로 설치 되어 있는 경우 아래와 같이 출력된다. 

 

또한 터미널에 아래와같이 입력하여 설치해볼수도 있다. 

sudo apt install nvidia-cuda-toolkit