본문 바로가기
코딩 라이프/개발환경 셋팅

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

by min0114 2022. 7. 26.

환경 셋팅 

 

우분투 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

Previous releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production

developer.nvidia.com

 

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

cuDNN provides researchers and developers with high-performance GPU acceleration.

developer.nvidia.com

 

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 

반응형

댓글