can’t convert cuda:0 device type tensor to numpy
测试代码：
import torchimport numpy as npinput_tensor = torch.rand(size=(1, 3, 416, 416)).cuda()bbb=np.array(input_tensor)
numpy 1.21报错,解决方法：
import torchimport numpy as npinput_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
详细的报错信息如下： RuntimeError: Expected all tensors to be on the same device, but fo
pytorch报错： IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a
使用TF进行计算，出现： Traceback (most recent call last): File "infer.py", line 14, in
解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim ten
一般有三点错误 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 1、
先单独创建一个py文件用于测试cuda是否配置正确 print(torch.cuda.is_available()) 运行后输出的是True，这代表你的cuda已经
can't convert cuda:0 device type tensor to numpy 测试代码： import torch i
can't convert cuda:0 device type tensor to numpy x是list，list内容是tensor cuda 代码： x = np
运行程序如下： import numpy as np import torch from torch import nn from torch
wu@wu-X555LF:~$ nvidia-smi Sat Jul 27 14:07:04 2019 +----------------
超、凢脫俗
心已赠人
柔情只为你懂
灰太狼
小鱼儿
还没有评论，来说两句吧...