Windows 11에 CUDA/cuDNN+Pytorch 설치

1. 준비 작업:

토치 버전 확인: Python 대화형 환경으로 들어갑니다.

>>>import torch
>>>torch.__version__

cuda 버전 확인: CMD 창

nvcc --version

 버전이 일치하지 않으면 제거하고 다시 설치해야 합니다.

2. 설치

Windows에 CUDA/cuDNN 설치 - Zhihu medium - Windows에 CUDA 설치: The Definitive Guidemedium - Windows 10에 CUDA 및 cuDNN 설치 Windows는 cudn과 cudnn 버전 간의 관계를 설치하고 구성합니다... https://zhuanlan.zhihu.com/ p/99880204 CUDA와 Pytorch 간의 통신 보기:

이전 PyTorch 버전 | PyTorch 연구 프로토타이핑에서 프로덕션 배포로의 경로를 가속화하는 오픈 소스 머신 러닝 프레임워크 https://pytorch.org/get-started/previous-versions/ 이 머신은 CUDA11.8이므로 설치 명령은 다음과 같습니다. 처음 세 개:

# ROCM 5.4.2 (Linux only)
pip install torch==2.0.0+rocm5.4.2 torchvision==0.15.1+rocm5.4.2 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/rocm5.4.2
# CUDA 11.7
pip install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117
# CUDA 11.8
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
# CPU only
pip install torch==2.0.0+cpu torchvision==0.15.1+cpu torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cpu

3. 검증

pytorch에 해당하는 cuda 버전 보기

python
>>>import torch
>>>torch.version.cuda

GPU를 사용할 수 있는지 확인

import torch
import torchvision
print(torch.cuda.is_available())