can‘t convert cuda:0 device type tensor to numpy

can't convert cuda:0 device type tensor to numpy

测试代码：

      import torch
      import numpy as np
       input_tensor = torch.rand(size=(1, 3, 416, 416)).cuda()
       bbb=np.array(input_tensor)
  
 

numpy 1.21报错,解决方法：

      import torch
      import numpy as np
       input_tensor = torch.rand(size=(1, 3, 416, 416)).cuda()
       input_tensor=input_tensor.cpu()
       bbb=np.array(input_tensor)
  
 

但是在Numpy1.19下载不报错，出来的bbb也在cuda0上。

pip install numpy==1.19

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/118501079