728x90
반응형
sample_tensor.is_cuda # gpu 텐서 확인 리턴값 bool
sample_tensor.to(device) # 설정한 디바이스로 텐서 타입 변경
sample_tensor.cuda() # 다른 방식 1
sample_tensor = sample_tensor.to(device. dtype=torch.float64) # 다른방식 2
sample_tensor = torch.as_tensor(sample_tensor, dtype=torch.float32, device='gpu') # 다른방식 3
sample_tensor.cpu() # CPU로 옮길 때
반응형
'데이터분석 > Pytorch' 카테고리의 다른 글
tensor 생략없이 (0) | 2024.05.03 |
---|---|
모델 학습 GPU 병행처리(nn.DataParallel) (0) | 2024.03.13 |
[Pytorch] 파이토치로 GPU 사용 (0) | 2022.11.28 |
[Pytorch] Tensor(텐서) 오류 해결 모음 (0) | 2022.11.28 |
[Pytorch] 파이토치 Tensor(텐서)의 최대 값, 인덱스 (1) | 2022.11.28 |
댓글