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

代码 报错信息 原因 看信息应该是说数据在显存里plt不能直接调用 所以要先复制到宿主内存里面 解决方法 倒数第二三行修改为: ...

2022-02-20 21:37 0 3466 推荐指数:

查看详情

张量与数组运算报错(Use Tensor.cpu() to copy the tensor to host memory first;RuntimeError: Expected all tensors to be on the same device)

结论:1.张量与数组运算,张量必须在cpu上,产生结果为cpu上的张量,可继续与数组运算(张量必须在gpu上)    2.张量与张量运算,cpu上的张量与gpu上的张量是无法运行的,必须在相同的gpu上或cpu上,猜想不同型号的gpu因该也不行。 一.张量与数组运算,前提张量 ...

Tue Mar 16 09:04:00 CST 2021 0 6880
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM