Pytorch, CUDA, cuDNN 설치 [windows 10]

 

 

1. 그래픽 카드 확인

1. 상태바에서 마우스 우클릭

2. 작업 관리자 클릭

3. 성능 클릭

4. GPU 클릭

GPU를 간단히 확인할 수 있다.

2. CUDA, cuDNN 설치

딥러닝 프레임워크에서 GPU를 사용하여 계산속도를 올리기 위해서는 NVIDIA GPU, NVIDIA 그래픽 드라이버, CUDA, cuDNN을 설치해야 한다. 그래픽 드라이버 설치는 어렵지 않음으로 패스

1. pytorch 홈페이지 접속

2. install 클릭

3. GPU와 호환되는 CUDA 버전을 확인해야 한다.

30 시리즈 이상은 CUDA 11 20 시리즈 아래는 10.2를 설치하면 웬만하면 호환된다. 필자는 3060ti이므로 CUDA 11.1을 설치

4. 구글에 cuda (버전) install 검색 후 홈페이지 접속

5. 아래 사진과 같이 선택한 후 다운로드 후 실행

6. 다음, 다음, 다음.. 마침

CUDA 설치가 완료되면 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1 폴더가 하나 생성되고 CUDA_PATH 환경변수가 추가된다.

7. 구글에 cuDNN install 검색하고 cuDNN 다운 버튼 클릭 후 버전에 맞춰 압축파일 다운

8. 압축해제 후 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1 폴더 안으로 bin, include, lib 폴더 이동

3. Pytorch 설치

필자는 LTS(Long Time Service)를 한번 설치하고 오래 사용하고 싶어서 LTS버전을 다운로드한다.

1. cmd 창을 열고 위의 이미지에 보이는 pip3 install 명령어를 실행한다.

4. GPU 사용 확인

python코드를 아래와 같이 작성한 후 실행하여 True가 출력되면 CUDA설치가 완벽하게 된 것이다.

import torch

print(torch.cuda.is_available())