Windows PyTorch 설치: CUDA 를 곁들인..

ChatGPT 의 안내를 받아서 콤푸타에 PyTorch 를 사용하게끔 만들어보자.

Windows 에서 CUDA 용 PyTorch 설치. 출처: YouTube CloudCasts - Alan Smith

---

Anaconda 내 Jupyter Notebook에서 사용하기 위해 CUDA 12.6 지원과 함께 PyTorch를 설치하려면 다음 단계를 따르세요.

1. Anaconda 프롬프트 열기

시스템에서 Anaconda Prompt를 열어 시작하십시오.

2. 새 Conda 환경 생성(선택 사항)

종속성 충돌을 피하기 위해 PyTorch 프로젝트에 대한 새로운 Conda 환경을 만드는 것이 좋습니다. 이름을 pytorch_env(또는 다른 이름)로 지정할 수 있습니다.

conda create --name pytorch_env python=3.12 -y
conda activate pytorch_env

3. CUDA 12.6과 함께 PyTorch 설치

PyTorch 공식 웹사이트에서 제공하는 PyTorch 설치 명령어를 사용하세요. CUDA 12.6은 PyTorch가 일반적으로 기본적으로 지원하는 것보다 최신 버전이므로 웹 사이트를 확인하거나 다음 명령을 시도하여 호환성을 확인해야 할 수도 있습니다.

PyTorch, TorchVision 및 TorchAudio를 설치합니다.

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

CUDA 12.6용 공식 빌드가 있는 경우 '12.4'을 '12.6'으로 바꾸세요(그러나 PyTorch CUDA 릴리스는 CUDA 툴킷 버전보다 약간 뒤처지는 경우가 많습니다).

4. Jupyter 노트북 설치

아직 Jupyter Notebook을 설치하지 않은 경우 환경 내에 설치하세요.

conda install -c conda-forge notebook

5. Jupyter Notebook에 환경 추가

새로운 Conda 환경을 Jupyter Notebook에 커널로 추가합니다.

conda install ipykernel
python -m ipykernel install --user --name pytorch_env --display-name "Python (PyTorch)"

6. 설치 확인

PyTorch와 CUDA가 제대로 설치되었는지 확인하려면 다음을 실행하여 Jupyter Notebook을 시작하세요.

 

jupyter notebook

New - "Python (PyTorch)" 을 선택하여 Notebook을 열고 다음 코드를 실행합니다.

import torch
print("PyTorch version:", torch.__version__)
print("CUDA available:", torch.cuda.is_available())
print("CUDA version:", torch.version.cuda)

CUDA를 사용할 수 있고 버전이 12.6(또는 호환 버전)과 일치하면 계속 사용할 수 있습니다!

문제 해결

• PyTorch 호환성: CUDA 12.6이 아직 지원되지 않는 경우 CUDA 12.4용 PyTorch 호환 버전을 설치하거나 CPU 전용 버전을 사용해야 할 수도 있습니다.
• 드라이버 업데이트: NVIDIA 드라이버가 CUDA 12.6을 지원하도록 업데이트되었는지 확인하세요.