[ Pytorch ] 파이토치 설치하기

---

[ Pytorch ] 파이토치 설치하기

https://pytorch.org/

---

 

머신러닝에서 tensorflow와 pytorch는 양대 산맥이죠

pytorch를 설치해봅시다.

 

 

 

https://pytorch.org/get-started/locally/

PyTorch

 

 

1. pytorch 버전

 

현재 1.9 까지 나왔네요.

이전버전으로도 설치가 가능합니다.

https://pytorch.org/get-started/previous-versions/

 

 

2. Package 

 

여러가지 방식으로 다운로드 가능한데요.

Conda를 사용하면 편리합니다.

물론 소스에서 직접 다운받아도 되구요.

 

 

3. Cuda

 

보통 nvidia gpu를 이용하실 겁니다.

gpu 사용을 위해서는 cuda 버전을 체크해야합니다.

 

본인의 nividia gpu 가 궁금하다면 아래의 코드를 통해 확인해 봅시다.

 

$ nvidia-smi

 

 

4. 설치하기

 

 

홈페이지에서 알려준 코드를 실행시킵니다.

$ conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

 

 

5. 설치 확인

 

 

python을 열어서 확인해봅시다.

아래의 결과를 보여준다면 잘 설치 된겁니다.

import torch
x = torch.rand(5, 3)
print(x)
torch.cuda.is_available()

print("PyTorch version: {}".format(torch.__version__))
print("CUDA version: {}".format(torch.version.cuda))
a = torch.randn(1, 10).to(device)
print(a)

---

[ 다음 글 ]

[ Pytorch ] 파이토치 텐서 합치는 방법 : cat(), stack() ( + dim의 의미와, 병합 방식의 차이)

[ Pytorch ] nn.BCELoss, nn.BCEWithLogitsLoss, nn.CrossEntropyLoss, nn.NLLLoss 총정리

[ PyTorch / torchvision ] make_grid() 사용하기

[ PyTorch / torchvision ] draw_bounding_boxes() 사용하기

---