[DL][Windows 11] 윈도우 파이토치 (PyTorch) 설치

0. 파이토치 (PyTorch) 설치 이전

0.1. 파이썬 설치

0.2. CUDA, cuDNN 설치

0.3. Python 가상환경 설정

위 3가지 중 1번과 2번은 필수이며, 3번은 선택입니다.

추가적으로 jupyter lab을 설치하시면 더욱 GUI 환경에서 쉽게 접근 가능합니다.

 

해당 내용에 대해서는 앞서 작성한 글인 "윈도우 11 텐서플로우 설치 편"과 "윈도우 CUDA, cuDNN 설치 편"을 참고 하시면 됩니다.

 

0.1. 파이썬 설치 및 Python 가상환경 설정

https://lonaru-burnout.tistory.com/17

[DL][Windows 11] 윈도우 텐서플로우 (TensorFlow) 설치

 

0.2. CUDA, cuDNN 설치

https://lonaru-burnout.tistory.com/16

[ML][Windows 11] CUDA, cuDNN 설치

 

---

 

1. PyTorch 설치

https://pytorch.org/get-started/locally/

PyTorch

 

해당 링크는 PyTorch 설치를 위한 Command code를 작성해줍니다.

 

 

본 글에서는 앞서 Conda가 아닌 Python의 even을 통해 가상환경을 설정했으므로, pip를 통해 설치합니다.

또한, CUDA 버전은 11.8로 설치 되어 있으므로 수정해줍니다.

옵션을 설정 한 후, 하단 "Run this Command:" 의 코드를 복사하여 command 창에서 실행합니다.

 

---

 

2. PyTorch 설치 확인

import torch
torch.cuda.get_device_name(0)
torch.cuda.is_available()
torch.__version__

위 코드를 입력하시면 순서대로 사용중인 gpu, cuda 활성화 여부, 파이토치 버전이 출력됩니다.