아래 링크를 통해 본인 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.html'Code > Pytorch' 카테고리의 다른 글
| [Pytorch] Batch 개수, Data 개수 (0) | 2023.05.14 |
|---|---|
| [Pytorch] Seed 고정 (0) | 2023.05.09 |
| [Pytorch] pytorch version 확인 (0) | 2023.04.14 |
| [Pytorch] Tensor 에서 특정 index 의 값을 뽑아 새로운 tensor 정의 (0) | 2023.04.12 |
| [Pytorch] .detach().cpu().numpy() (0) | 2023.04.10 |