Windows 에 CUDA 11.8과 PyTorch를 설치하는 방법을 설명합니다.
2023. 5. 7 최초작성
2023. 9. 9
2024. 1. 7 별도의 CUDA 없이 설치
2024. 3. 16 NVIDIA 그래픽 카드 드라이버 설치 방법 분리
글 업데이트
2025. 1. 27 글 업데이트
2025. 4. 25 NVIDIA 그래픽카드 설치방법 생략 및 글 업데이트
NVIDIA 그래픽 카드를 사용하는 경우를 가정하고 진행합니다.
PyTorch 설치
1. Miniconda를 설치 후, Visual Studio Code와 같이 사용하는 것을 권장합니다.
Visual Studio Code와 Miniconda를 사용한 Python 개발 환경 만들기( Windows, Ubuntu, WSL2)
2. 아래 링크에 접속하여 Your OS는 Windows, Package는 파이썬 환경에 따라 Conda 또는 Pip, Language는 Python, Compute Platform은 CUDA 11.8 또는 CUDA 12.6 또는 CUDA 12.8를 선택하면 아래쪽에 명령어가 보입니다.
현재 2025년 4월 25일 기준으로 보이는 선택항목이며 나중에 바뀔 수 있습니다.
https://pytorch.org/get-started/locally/
위 링크에 방문시 원하는 CUDA 버전을 선택하여 생성된 명령을 사용하여 설치를 진행합니다. Package 항목은 이젠 pip만 지원합니다.
여기에선 CUDA 12.6을 선택한 경우를 진행합니다.
윈도우 키를 누르고 anaconda를 입력하여 검색된 Anaconda Prompt (miniconda3)를 실행합니다.
PyTorch를 위한 가상환경을 생성합니다. 포스트에서는 Python 3.9 이상을 사용하는 파이썬 가상환경을 만듭니다.
(base) C:\Users\webnautes>conda create -n pytorch python=3.10
y입력후 엔터를 누릅니다.
Proceed ([y]/n)?
가상 환경을 활성화합니다.
(base) C:\Users\webnautes>conda activate pytorch
(pytorch) C:\Users\webnautes>
앞에서 확인한 다음 명령을 사용하여 CUDA 12.6을 사용하는 파이토치를 설치합니다.
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
설치완료후 done이 출력됩니다.
3. 파이썬 인터프리터에서 PyTorch가 GPU를 사용할 수 있는지 체크합니다.마지막에 True가 나와야 합니다.
(pytorch) C:\Users\webnautes> python -c "import torch; print(torch.cuda.is_available())"
True
Member discussion