본문 바로가기
Python

Pytorch 설치

by 187cm 2024. 4. 12.
반응형

 

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()

 

설치 끝

 

 

 

반응형