Windows 10에 CUDA, Pytorch 설치하기

1.설치 환경

윈도우 10

2. CUDA 설치하기

CUDA Toolkit 12.1 Downloads

 

 

위 페이지 접속해서 아래와 같이 클릭후 .exe 파일 설치

 

다음과 같은 화면이 나옴 (경로설정)

 

설치가 끝나고 프롬프트에서 nvcc --version 명령어를 치면 정상적으로 cuda가 설치된 것을 알 수 있다.

 

3. Pytorch 설치하기

아래 사이트 접속

Start Locally

 

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

 

 

 

정상적으로 설치된 것을 알 수 있다.