본문 바로가기
에러 해결법

윈도우 5090 지원하는 pytorch 버전

by 블로그별명 2025. 5. 17.

2025-05-17 기준입니다

# 기존 환경 제거 후 (선택 사항)
pip uninstall torch torchvision torchaudio

# CUDA 12.8 지원 버전으로 설치 (Windows 호환)
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

 

우분투의 경우

pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128

댓글