Python
Pytorch 설치
187cm
2024. 4. 12. 12:43
반응형
1. MobaXterm 설치해서 원격 서버 접속 후 버전 확인
2. https://pytorch.org/get-started/previous-versions/ 에서 맞는 버전으로 설치
- 11.4는 없으니까 그 아래 버전인 11.3으로 검색
- 필자는 +cu 가 붙은걸 선호하니 그걸로 복사 후 설치.
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
import torch
torch.cuda.is_available(), torch.cuda.device_count()
설치 끝
반응형