[Python] CUDA 설치 및 연결

1. GPU 확인

장치관리자 > 디스플레이 어댑터에서 GPU 모델명 확인

 

 

2. 엔비디아 드라이버 다운로드 & 설치

내 GPU 모델을 선택해서 드라이버 다운로드 & 설치

ㅌ

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

Download the latest official NVIDIA drivers

 

3. CUDA 설치

3.1 내 GPU의 Computing capability 확인하고 그에 맞는 CUDA 버전 확인

: 저의 경우는 computing capability 6.1 -> cuda 12.1 다운로드

https://en.wikipedia.org/wiki/CUDA#GPUs_supported

CUDA - Wikipedia

 

 

3.2 구글에 CUDA Toolkit xx.x 검색해서 다운로드 & 설치

 

3.3 설치 완료 후 환경변수 확인

시스템 > 정보 > 고급 시스템 설정 > 고급 > 환경 변수 

 - 시스템 변수에 CUDA_PATH, CUDA_PATH_Vxx_x 추가 되었는지 확인

 

- Path 더블 클릭, 아래 경로 추가되었는지 확인

 

 

4. cuDNN 다운로드

4.1 CUDA에 맞는 버전으로 다운로드

https://developer.nvidia.com/cudnn-downloads#a-collapse714-92

cuDNN 9.1.0 Downloads

 

4.2 압축 풀기 후 환경변수 경로에 다운로드 파일 넣어주기

C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/vxx.x에 bin, include, lib 파일 추가해주기

 

 

여기까지하면 GPU 사용을 위한 CUDA 설치 완료~

근데 torch.is_available()로 안 불러와 진다면, torch 버전 확인해주기! 

 

* 최종

 - Python : version 3.11.9

 - NVIDIA Driver : GeForce GT 1030 (computing capability : 6.1)

 - CUDA :  version 12.1

 - cuDNN : version 8.9.7 for cuda 12.x

 - pytorch : version 2.2.2

pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu121