CUDA 설치

참조
https://teang1995.tistory.com/17

그래픽카드 확인

내 글카 검색해 Compute Capability 확인
https://www.wikiwand.com/en/CUDA

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

에 접속 후, 본인의 GPU의 Compute Capability에 맞는 CUDA SDK 버전을 확인. 12버전을 사용해도 될거 같다

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local
여기 들어가서 다운로드 후 설치

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA #설치경로

cuDNN 설치
https://developer.nvidia.com/rdp/cudnn-download

https://pytorch.org/

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

import torch
torch.cuda.is_available()