인공지능을 좋아하는 곧미남

gpu에 할당된 tensor데이터를 numpy array로 변환할때 cpu 연산으로 변환한 뒤 실시 본문

code_study/pytorch

gpu에 할당된 tensor데이터를 numpy array로 변환할때 cpu 연산으로 변환한 뒤 실시

곧미남 2022. 5. 16. 10:19

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

input.float().to(device)

np.array(input[i].permute(1, 2, 0).cpu(), dtype=np.float32)

이때 input에 .cpu()로 cpu연산 변환이 되지 않으면, 아래 Error가 발생함.

 

TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

 

저작자표시 비영리 동일조건

'code_study > pytorch' 카테고리의 다른 글

[torchvision] ImageFolder  (0) 2022.05.16
pytorch의 def forward(self, x)  (0) 2022.01.18
BackBone Encoder Layer에서 Feature Map 추출  (0) 2022.01.17
torch.nn.Sequential  (0) 2022.01.14
pytorch DataLoader  (0) 2022.01.10
'code_study/pytorch' Related Articles more
Comments