본문 바로가기
기타 꿀팁

Cuda & Cudnn 설치

by winston1214 2020. 12. 16.
반응형

## 환경

Ubuntu18.04.5 LTS, GPU : 2080 Super

 

Linux를 너무 많이 깔아서.. 이제 외울정도로 됐지만.. 정말 아직도 가끔씩 헷갈리고 맨날 구글링을 하다가 이제 정리한다.

 

일단 체크를 해야될 것은 nvidia driver를 깔기 전에 HDMI 또는 DP 선이 어디에 꽂혀있는지를 확인해야 된다!! 

아무것도 모르고 본체 아무 곳에 꽂았다가...지옥을 맛보고 내 데이터들을 날린 적이 한두번이 아니다...

 

그리고 꿀팁!!! cuda 설치 시 nvidia driver 따로 설치 안해도 된다!!!! (그 누구도 안알려줬는데 cuda 설치 10번만에 깨달음..)

 

이제..드디어 cuda를 설치해보겠다.

- cuda 설치

developer.nvidia.com/cuda-downloads

 

CUDA Toolkit 11.2 Downloads

Select Target Platform Click on the green buttons that describe your target platform. Only supported platforms will be shown. By downloading and using the software, you agree to fully comply with the terms and conditions of the CUDA EULA. Operating System

developer.nvidia.com

 

공식 홈페이지에 들어간다! 벌써 11.2까지 나왔다...하지만 모든 최신은 너무 안좋으므로..난 10.2를 설치하겠다.

자기 사양에 맞는 것을 선택한 후 친절하게 나와있는대로 따라하면 된다.

나는 Ubuntu 18.04 x86에 deb(local)로 선택하였다.

$ 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/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get -y install cuda

 

세번째 줄에서 오래걸릴 것이다. 만약에 출장을 나간다던가 빠른 설치를 해야된다면 세번째 줄에 있는 것을 미리 다운받고 그 컴퓨터에 설치하면 된다.

만약 설치할 때 E: Unmet dependencies. Try ‘apt fix-broken install’ with no packages ( or specify a solution) 라는 에러가 뜨면

sudo apt-get -o Dpkg::Options::="--force-overwrite" install --fix-broken

해준다.

그리고 환경 변수 설정을 한다. (sudo gedit ~/.profile 로 접근해도 되고 아래와 같은 명령어 삽입도 가능)

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

 

이를 끝내고 sudo reboot를 꼭!!! 해준다! 정말 만약에 reboot를 하는 과정에서 에러가 나면 정말 .. 그 전에 내가 강조했던 것들을 다시 한 번 체크하기 바란다. (nvidia driver가 이미 설치되어 있으면 driver 버전이랑 cuda 버전이랑 안맞을 수 있다.)


1. Ubuntu 재설치

2. 다 ok, ok, ok 나오고 로그인 화면으로 안넘어 간다면 ctrl + alt + f2 를 누르고(cli모드) 아이디 비번 누르고 sudo apt purge nvidia* 하고 쏵 다 삭제 후 다시 설치


그러고 잘 깔렸는지 확인은 nvidia-smi , nvcc -V 를 통해 확인을 할 수 있다.

 

- cudnn 설치

developer.nvidia.com/rdp/cudnn-download#a-collapse765-102

 

nvidia 홈페이지에 로그인을 해야된다.(구글로 간단하게 로그인 가능)

그러고 자신의 cuda version과 맞는 cudnn의 압축 파일을 다운 받는다.

그리고 다운 받은 파일의 폴더로 이동 한 후 작업한다.(cd downlodas)

$ tar –xzvf cudnn-탭
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

 

cudnn tar.gz 파일을 다 치긴 귀찮으니 탭을 일상화 하자! 일부러 버전이 다를까봐 탭으로 적어놨다.

위 작업이 성공적으로 끝났다면 cudnn 설치가 완료된 것이다!

 

진짜 ubuntu를 몇번이고 엎어도 이 블로그를 통해 cuda 와 cudnn은 가볍게 설치했으면 좋겠다...

 

 

반응형

'기타 꿀팁' 카테고리의 다른 글

Ubuntu USB 설치  (0) 2021.07.08
동영상 프레임 단위로 자르기  (0) 2021.04.14
jetson nano opencv webcam  (3) 2021.03.23
동영상 gif 변환  (0) 2020.12.23
Jetson AGX Xavier 초기화  (2) 2020.11.06

댓글