碼上快樂

相關內容    簡體    繁體

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

本文轉載自   查看原文   2019-04-09 15:46   3398    python

報錯原因：numpy不能讀取CUDA tensor 需要將它轉化為 CPU tensor。

所以如果想把CUDA tensor格式的數據改成numpy時，需要先將其轉換成cpu float-tensor隨后再轉到numpy格式

報錯行：

tcls[index, best_n, g_y_center, g_x_center, np.array(target[index, t, 0])] = 1

修改后：

tcls[index, best_n, g_y_center, g_x_center, np.array(target[index, t, 0].cpu())] = 1

 


免責聲明！

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。



猜您在找 TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first. can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first. TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory .. 【E-13】TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first. 張量與數組運算報錯(Use Tensor.cpu() to copy the tensor to host memory first;RuntimeError: Expected all tensors to be on the same device) 2、tensor數據的運算、numpy與Tensor互轉、Tensor轉cuda TypeError: Can not convert a float32 into a Tensor or Operation. tensorflow TypeError: Can not convert a float32 into a Tensor or Operation TypeError: Fetch argument 0.484375 has invalid type , must be a string or Tensor. (Can not convert a float32 into a Tensor or Operation.) IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item ()` in C++ to convert a 0-dim tensor to a number
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM