CUDA, CUDNN install. 우분투 쿠다 설치

1. Cuda 다운로드

 

developer.nvidia.com/cuda-toolkit-archive

CUDA Toolkit Archive

 

설치하고자 하는 CUDA 버전을 선택합니다.

 

다음은 설치 옵션입니다.

 

설치 옵션을 선택하면 다음과 같이 설치 파일을 다운로드 할 수 있는 코드가 주어집니다.

 

해당 코드는 현재 디렉토리에서 설치되기 때문에 설치되는 위치에 주의해주시기 바랍니다.

(터미널을 실행하여 다음 코드를 실행하면 cd를 했을 때의 위치인 /home/user 위치에 설치됩니다.)

 

2, 설치

 

해당파일을 다운로드한 후 해당 폴더에서

sudo sh cuda_xx.x.x_xxx.xx.xx_linux.run

을 입력해서 설치를 진행합니다.

 

첫번째는 동의여부를 확인합니다.

accept를 이용하여 넘어가줍니다.

 

두번째는 선택입니다.

그래픽 드라이버는 설치가 되어 있으므로 Enter(엔터)를 눌러 취소를 한 후 CUDA를 설치합니다.

 

3. 환경 변수 설정

 

다음을 입력하여 환경 변수 설정에 들어갑니다.

sudo vim ~/.bashrc

a나 i를 눌러 입력모드로 들어간 후 다음을 입력하고 Esc를 누른 후 :WQ를 입력하여 저장 후 빠져나옵니다.

export PATH=/usr/local/cuda-11.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

다음으로 

source ~/.bashrc

를 입력하면 완료가 됩니다.

 

이후 명령어를 통해 설치 여부를 확인할 수 있습니다.

nvcc -V

 

 

4. cuDNN 다운로드

 

developer.nvidia.com/cudnn

NVIDIA cuDNN

홈페이지에 접속하여 Download를 누릅니다.

 

그러면 로그인을 하라고 하는데 아이디가 없다면 가입을 한 후 다운로드를 진행해줍니다.

(구글 계정으로 하시면 편합니다.)

 

가입한 후 다운로드를 누르면 다음과 같은 화면이 뜨는데 젤 아래쪽에 Releases를 클릭하여 이전 버전의 cuDNN 또한 설치가 가능합니다.

선택 후 해당 운영체제에 맞는 파일을 다운로드합니다.

해당 파일은 웹페이지에서 다운로드가 진행되므로 Downloads 폴더로 이동됩니다.

따라서 설치를 진행하실 때 해당 파일이 있는 Downloads 폴더에서 진행하면 됩니다.

 

먼저 해당 압축 파일을 풀어준 후 cuda 파일에 옮겨줍니다.

tar -zxvf cudnn-xx.x-linux-x64-vx.x.tgz
## cuda에 파일 옮기기
sudo cp cuda/include/cudnn.h /usr/local/cuda-xx.x/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-xx.x/lib64

이렇게 하면 설치가 완료되었습니다.

 

여러 CUDA 버전을 이용할 때에는 CUDA와 CuDNN을 설치 한 후 아까 설정한 export 환경변수 부분 폴더 경로를 수정하면 다른 CUDA 버전을 이용할 수 있습니다.