Windows 10 PyTorch 설치

Windows에서 CUDA를 활용한 NVIDIA GPU 작업용 PyTorch 설치

*자세한 설명 생략

 

 

▷ NVDIA GPU 드라이버 확인

*NVDIA GPU 드라이버가 설치돼 있어야 GPU를 활용

*설치 방법 생략(아래 참고 링크)

설치확인

 

 

▷ CUDA 11.8 Toolkit 설치

*CUDA: NVIDIA의 GPU 연산을 실행하고 관리하는 기술

*설치 방법 생략(아래 참고 링크)

*cuDNN은 PyTorch 설치 시 자동으로 포함(미설치)

설치 확인

 

 

▷ PyTorch 설치

*GPU 연산을 위해 CUDA를 사용하는 라이브러리

torch 2.5.1
*Python 3.12.2, CUDA 11.8과 호환

--index-url https://download.pytorch.org/whl/cu118
*CUDA 11.8 환경에 최적화된 PyTorch 패키지 다운로드 경로


💡 pip install torch==2.5.1 --index-url https://download.pytorch.org/whl/cu118

설치 확인

 

 

▷ 참고

NVIDIA 그래픽 드라이버 다운로드 방법, 그래픽 드라이버 잡는 법, 지포스 4060

Window에서 CUDA 11.8, cuDNN 8.7.0 설치