[Pytorch] 텐서(Tensor) 자료형

쿠다 설치

https://developer.nvidia.com/cuda-downloads

CUDA Toolkit 12.1 Downloads

버전 선택을 잘 해야 합니다

지난 버전을 선택하고 싶다면 빨간 네모 박스 클릭

 

쿠다 버전에 맞게 터미널에서 파이토치 설치

https://pytorch.org/

PyTorch

자신의 컴퓨터 상황에 맞게 체크를 하고 command를 복사합니다.

 

 

cmd창에 붙여넣기

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

 

 

---

<파이토치 사용하기>

 

파이토치 라이브러리 임포트

import torch

 

현재 파이토치 버전 확인

torch.__version__

 

그래픽 카드(GPU) 사용 되는 지 확인

torch.cuda.is_available()

 

그래픽 카드 개수 확인

torch.cuda.device_count()

 

그래픽 카드 이름 확인

# 번호 쓰기

# 0 번째 device 이름 확인
torch.cuda.get_device_name(0)

 

tensor 자료형 만들기

# scalar
torch.tensor(5)

# vector
torch.tensor([1,2,3])

# matrix
torch.tensor([[1,2,3], [4,5,6]])

파이썬 list to tensor 가능

 

numpy to tensor

# numpy to tensor
import numpy as np

data = np.array([1,2,3])
torch.tensor(data)

 

tensor 요소를 다 1로 채우기

# shape을 넣어준다

# 1차원 2
torch.ones(2)

# 2차원 2X3
torch.ones(2, 3)

# 3차원 3X128X128
torch.ones(3, 128, 128)

 

tensor 요소를 다 0으로 채우기

# shape을 넣어준다

torch.zeros(2)

torch.zeros(2, 3)

torch.zeros(3, 128, 128)

 

tensor 단위 행렬 만들기

torch.eye(4,4)

torch.eye(5) # torch.eye(5,5)와 동일

 

arange([start], [stop], [step])

torch.arange(2, 10, 2) # tensor([2, 4, 6, 8])

 

0~1 사이 랜덤한 난수를 가진 tensor 만들기

# shape를 넣어준다

torch.rand(5)

torch.rand(3,3)

torch.rand(3, 128, 128)