Data Science/Anaconda

RTX3070 GPU 사용하기 (CUDA 11.2, cuDNN 설치) for Windows

RTX3070에서 GPU를 사용하는 사용하기 위해 CUDA와 cuDNN 환경 세팅을 해보자.

 

 

CUDA 설치하기

CUDA 설치를 위해 아래 링크로 이동하자.

 

 

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

 

아래와 같은 링크가 나오면 11.2.x 버전을 클릭한다.

 

 

운영체제에 맞게 플랫폼을 선택하고 Download 버튼을 클릭한다.

 

 

실행 파일을 실행시키고 OK를 눌러주자.

 

 

설치 준비가 되면 아래 창이 뜰 것이다.

동의 및 계속을 누른다.

 

 

특별한 경우가 아니라면 설치 옵션은 그대로 두고 다음을 누른다.

 

 

Nsight를 위한 Visual Studio가 설치되어 있는지 확인하는 것이다.

굳이 필요하지 않다면 체크 박스를 체크하고 NEXT를 누른다.

 

 

다운로드가 진행된다.

 

 

다운로드 요약을 보여준다. 확인 후 다음을 누른다.

 

 

설치가 완료되었다.

 

 

 

cuDNN 설치

cuDNN 설치를 위해 아래 링크로 이동한다.

 

cuDNN Archive

NVIDIA cuDNN is a GPU-accelerated library of primitives for deep neural networks.

developer.nvidia.com

 

CUDA 버전에 맞는 cuDNN을 설치하면 된다.

나는 8.1.1 버전을 다운로드했다.

 

 

Zip 파일로 다운로드한다.

설치를 위해 로그인 과정이 필요하다. NVIDIA 계정이 없다면 회원가입 하자.

 

 

설치가 완료됐으면 CUDA가 설치된 경로에 아래와 같이 압축을 풀어준다.

설치 경로는 "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2" 이다.

 

 

 

 

설치가 제대로 되었는지 확인하기 위해 cmd를 열어 아래 명령어를 입력해보자.

nvcc --version