파이토치 device (gpu / cpu) 관련 질문드립니다.
711
1 asked
파이토치를 처음 배우는 학생입니다.
처음 텐서를 만들 때 cpu와 gpu 중 원하는 곳에 텐서를 저장할 수 있다고 배웠습니다.
이때 기본적으로 torch.tensor로 텐서를 만들면 cpu로 가는지, 아니면 cpu / gpu 그 어느것도 아닌 기본형으로 만들어지는지 궁금합니다.
또 이와 관련하여 텐서를 출력했을 때 <device = 'cuda:0'>이런 게 안 뜨면 cpu에 저장되어 있다고 보면 되는건가요?
Answer 1
0
device를 지정해 주지 않고 만들면 CPU tensor가 되고 device를 "cuda:0"로 지정해 주고 만들면 GPU tensor가 됩니다.
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.int32)
x --> cpu tensor
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.int32, device="cuda:0")
x --> gpu tensordocument에 다음과 같이 설명이 되어 있습니다.
torch.tensor(data, *, dtype=None, device=None, requires_grad=False, pin_memory=False)
device (
torch.device, optional) – the device of the constructed tensor. If None and data is a tensor then the device of data is used. If None and data is not a tensor then the result tensor is constructed on the CPU.
더 자세한 사항은 https://pytorch.org/docs/stable/generated/torch.tensor.html 문서를 참조하세요. 좋은 질문 감사 드립니다.
import torch가 안되는 경우는 어떻게 하나요?
0
16
1
소리가 겹쳐서 들려요
0
20
2
20강에서 파인튜닝 때 사용한 데이터가 없어졌습니다. LoRA Trainer 매개변수도 라이브러리 업그레이드로 수정되었습니다.
0
20
1
[개정판] 딥러닝 컴퓨터 비전 완벽 가이드 먼저? 구현하며 배우는 Transformer 먼저?
0
30
1
수업자료
0
23
2
전 강의와 전혀 이어지지가 않음
0
30
1
pytorch local 설치 옵션에 conda 가 없습니다.
0
38
3
pc에서는 괜찮은데 탭으로 들으니 화면확대시 화면이 까맙니다
0
22
1
강의 환경설정 질문
0
40
2
모든 자료 다운로드 누를때마다 똑같은 excel파일이 다운로드 받아짐. 노션 주소 공유되나요?
0
32
2
오토인코더+ Knn, SVC 로 해석하는경우
0
45
3
강의자료에 소스코드가 없는데요
0
49
3
강화학습저장 및 로드
0
57
1
Custom Dataset에서의 polygon 정보 관련
0
83
3
paperswithcode 서비스 종료 관련 문의
0
209
2
mini-batch gradient descent 시 backpropagation에 대하여
0
75
2
이미지가 출력되지 않습니다
0
209
2
IMDBDataset 파일 다운로드가 안되는데 어떻게 받을 수 있을까요?
0
292
1
the kernel appears to have died. it will restart automatically.
0
666
1
nn.CrossEntropyLoss() method 질문
0
561
1
5분10초 경, 강의자료에 대하여 질문드립니다.
0
323
1
update 공식 관련 질문 (3분31초 경)
0
309
1
종속변수와 독립변수
0
578
1
linspace(100, 200, 10)
0
354
1