반응형
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()
설치 끝
반응형
'Python' 카테고리의 다른 글
timm 모델 리스트 + 옵티마이저까지 (timm model list / 2024.04.19) (2) | 2024.04.19 |
---|---|
Docker container에서 아나콘다 실행 및 기본 설정 (0) | 2024.02.27 |
리스트 안에 numpy array나 Tensor 배열이 갇혀 있을 때 해결방법. (0) | 2023.03.28 |
Python numpy 지수표기, e+ 표기 없이 출력하기 (0) | 2022.09.05 |