2021. 7. 5. 19:35ㆍ딥러닝
먼저 python과 pytorch가 설치되어있다는 가정하에 진행 합니다.
1. 본인의 컴퓨터에 그래픽 카드가 어떤건지, CUDA를 지원하는지 확인한다.
윈도우 검색창에 장치 관리자를 검색
2. 접속하여 본인의 그래픽 카드가 호환이 되는지 확인
https://developer.nvidia.com/cuda-gpus
CUDA GPUs
Your GPU Compute Capability Are you looking for the compute capability for your GPU, then check the tables below. NVIDIA GPUs power millions of desktops, notebooks, workstations and supercomputers around the world, accelerating computationally-intensive ta
developer.nvidia.com
3. 접속하여 툴킷을 설치한다
https://developer.nvidia.com/cuda-toolkit
CUDA Toolkit
CUDA Toolkit Develop, Optimize and Deploy GPU-Accelerated Apps The NVIDIA® CUDA® Toolkit provides a development environment for creating high performance GPU-accelerated applications. With the CUDA Toolkit, you can develop, optimize, and deploy your appl
developer.nvidia.com
설치 후에
import torch
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(device)정상적으로 설치가 되어있다면
cuda 라고 프린트 된다.