CUDA + cuDNN을 설치하고 PyTorch에서 GPU를 인식하는 단계 (Window11)

2022-10-03 last update

7 minutes reading

개요

WIndows11에 CUDA + cuDNN을 설치하고,
PyTorch로 GPU를 인식할 때까지의 순서 정리.

환경

OS : Windows11
GPU : NVIDIA GeForce RTX 3080 Ti

설치

CUDA Toolkit 설치

아래 링크에서 최신 (11.7) CUDA Toolkit을 다운로드하십시오.
CUDA Toolkit 11.7 Downloads

다운로드가 완료되면 설치 시작.
설치시의 설정은 모두 디폴트.

설치 완료 후, 환경 변수로 패스가 통과하고 있는지를 확인.

아래 명령을 실행하여 성공적으로 설치되었는지 확인.

nvcc -V

# nvcc: NVIDIA (R) Cuda compiler driver
# Copyright (c) 2005-2022 NVIDIA Corporation
# Built on Tue_May__3_19:00:59_Pacific_Daylight_Time_2022
# Cuda compilation tools, release 11.7, V11.7.64
# Build cuda_11.7.r11.7/compiler.31294372_0

cuDNN 설치

아래 링크에서 CUDA와 같은 버전에 해당하는 cuDNN을 다운로드.
cuDNN Download
다운로드가 완료되면 zip 파일을 압축 해제하고 내용을 임의의 폴더로 이동합니다.
※이번은 「C:/tool/cuda」에 넣었다.

폴더 이동 후 경로를 통과합니다.

다음 명령으로 설치 확인.

where cudnn64_8.dll

# C:\tools\cuda\bin\cudnn64_8.dll

PyTorch로 동작 확인

Anaconda에 환경 구축

아래의 명령으로 확인용 가상 환경을 Anaconda에 작성.

conda create -n pytorch python=3.7

conda activate pytorch

PyTorch 설치

아래 링크에서 자신의 환경에 맞는 명령을 조사하여 PyTorch 설치.
PyTorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

GPU 인식 확인

이하의 코드로 PyTorch로부터 GPU의 인식을 확인할 수 있다.
「True」가 되면 성공.

import torch
torch.cuda.is_available()

# True

Menu