
[PyTorch] CUDA, cuDNN 설치
·
PyTorch
PyTorch 설치 전에, CUDA를 먼저 설치한다. CUDA(Compute Unified Device Architecture)는 NVDIA가 제공하는 병렬 컴퓨팅 플랫폼 및 API로, GPU를 사용해 연산을 가속화하는데 사용된다. Pytorch와 같은 딥러닝 프레임워크에서 모델 훈련 및 추론을 가속화하기 위해 사용한다. 우선 GPU를 확인한다. 필자의 경우는 NVIDIA GeForce RTX 4050 Laptop GPU가 설치되어 있어 이를 기준으로 진행한다. https://developer.nvidia.com/cuda-downloads CUDA Toolkit 12.1 DownloadsGet the latest feature updates to NVIDIA's proprietary compute s..