Code & Framework/Pytorch
[Pytorch] GPU에 맞는 CUDA version 설치
이성훈 Ethan
2023. 4. 14. 04:54
728x90
아래 링크를 통해 본인 GPU 에 맞는 CUDA 버전 확인
https://www.wikiwand.com/en/CUDA#GPUs_supported
Wikiwand - CUDA
CUDA is a proprietary and closed source parallel computing platform and application programming interface that allows software to use certain types of graphics processing units for general purpose processing, an approach called general-purpose computing on
www.wikiwand.com
아래 링크를 통해 특정 CUDA 버전의 pytorch 다운로드
https://pytorch.kr/get-started/previous-versions/
파이토치 한국 사용자 모임 (PyTorch Korea User Group)
파이토치 한국 사용자 모임에 오신 것을 환영합니다. 딥러닝 프레임워크인 파이토치(PyTorch)를 사용하는 한국어 사용자들을 위해 문서를 번역하고 정보를 공유하고 있습니다.
pytorch.kr
예시) 아래와 같은 형태로 다운로드
pip install torch==1.12.1+cu113 torchvision==0.13.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html728x90