본문 바로가기
Programming/Setting

Windows10에 CUDA, cuDNN 설치하기

by EDGE-AI 2021. 7. 15.

1. CUDA

CUDA ("Compute Unified Device Architecture", 쿠다)는 그래픽 처리 장치(GPU)에서 수행하는 (병렬 처리) 알고리즘을 C 프로그래밍 언어를 비롯한 산업 표준 언어를 사용하여 작성할 수 있도록 하는 GPGPU 기술이다.

 

사용하고 있는 GPU에 맞는 CUDA 버전을 확인해줍니다.

en.wikipedia.org/wiki/CUDA#GPUs_supported

 

CUDA - Wikipedia

From Wikipedia, the free encyclopedia Jump to navigation Jump to search Parallel computing platform and programming model CUDA (Compute Unified Device Architecture) is a parallel computing platform and application programming interface (API) model created

en.wikipedia.org

제가 사용하는 GPU는 GTX 1660 Super이므로 Compute capabality는 7.5입니다.

Compute capabality 7.5는 CUDA10~11버전을 모두 지원합니다. 이 범위안에서 설치하시면 됩니다.

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

원하는 버전의 프로그램 다운로드 후 default 설정 그대로 설치하시면 됩니다. 저는 11.0 버전을 설치했습니다.

이런 폴더가 만들어지면 설치 완료입니다.

2. cuDNN

cuDNN은 엔비디아 CUDA 딥 뉴럴 네트워크 라이브러리, 즉 딥 뉴럴 네트워크를 위한 GPU 가속화 라이브러리의 기초 요소로 컨볼루션(Convolution), 풀링(Pooling), 표준화(Nomarlization), 활성화(Activation)와 같은 일반적인 루틴을 빠르게 이행할 수 있도록 하는 라이브러리입니다.

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

위 사이트에서 Download cuDNN -> Login -> License agree 후 CUDA 버전에 맞는 cuDNN을 받으시면 됩니다.

압축파일 내에 있는 파일들을 CUDA가 설치되어있는 폴더(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0)에 복사해줍니다.

3. 시스템 환경변수 확인

제어판 -> 시스템 -> 고급 시스템 설정 -> 고급 -> 환경 변수에 들어가서 아래와 같이 설정되어있는지 확인해줍니다.

되어있지 않은 경우 cuDNN을 복사한 폴더 경로로 추가하면 됩니다.

 

4. 설치 확인

nvidia-smi

위 CUDA 버전은 설치가능 버전이므로 신경쓰지 않으셔도 됩니다.

nvcc -V

여기까지 완료되면 설치 완료입니다.

 

이후 nvidia driver까지 설치해주시면 잘 사용할 수 있습니다!

www.nvidia.co.kr/Download/index.aspx?lang=kr

 

NVIDIA 드라이버 다운로드

 

www.nvidia.co.kr

 

댓글