티스토리 뷰

Skills/Error

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

미남잉 2023. 9. 18. 14:52
728x90

해당 에러의 발생 원인은 'cuda:0' device 타입의 tensor를 직접 numpy로 변환할 수 없어서 생기는 오류입니다. 이미 tensor는 gpu에 올려져서 계산되다가, numpy 연산을 하려면 cpu에서 가능하니 이런 에러가 발생하는 것입니다. 따라서 'Tensor.cpu()' 등의 메서드를 사용해야합니다.

해결 방법은 tensor를 cpu에 재할당 해주면 됩니다. 

 

보통 방법은 변수 뒤에 

.cpu().numpy()
.detach().numpy()

 

붙여주면 되는데 안 돼서 이번엔 이렇게 했더니 됐습니다.

 

.detach().cpu().numpy()

 

- detach(): 텐서를 다른 센터로부터 분리(detach)하여 연산 그래프에서 해당 텐서와 연결을 끊는 역할을 합니다. gradient 계산에서 분리할 때도 사용합니다.

 

 

 

728x90
저작자표시 비영리 변경금지

'Skills > Error' 카테고리의 다른 글

[Error] TensorBoard logging requires TensorBoard with Python summary writer installed. This should be available in 1.14 or above  (0) 2023.09.22
[Error] Python gprc AttributeError: 'NoneType' object has no attribute 'message_types_by_name'  (0) 2023.09.18
RuntimeError: Boolean value of Tensor with more than one value is ambiguous  (0) 2023.09.18
AttributeError: 'Namespace' object has no attribute 'learning_rate'  (0) 2023.09.18
RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /opt/conda/conda-bld/pytorch_1570910687230/work/aten/src/THC/THCGeneral.cpp:50  (0) 2023.09.09
댓글