首页 > 其他分享 >深度学习常见问题汇总

深度学习常见问题汇总

时间:2022-12-19 00:33:12浏览次数:33  
标签:常见问题 tensor 汇总 validity label cuda 深度 GPU numpy

1. Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

原因

参与运算的两个或多个变量,有的在CPU上,有的在GPU上

解决

首先找到报错的行,看看计算时都用到哪些变量或者数据,使用.is_cuda这个属性去查看到底哪些是在GPU上,然后统一放在一个设备上

print("validity_label is "+str(validity_label.is_cuda))
# 将validity_label放在cuda上
validity_label = validity_label.cuda()
print("validity_label is " + str(validity_label.is_cuda))

2.Tensor for argument #2 ‘mat1’ is on CPU, but expected it to be on GPU (while checking arguments for addmm)

原因

为了使模型在GPU上进行计算,需要将变量和模型都增加.to(device),都搬到GPU上去即可

解决

与错误1一样,将变量和模型搬到GPU上

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

原因

把CUDA tensor格式的数据改成numpy时,需要先将其转换成cpu float-tensor随后再转到numpy格式。 numpy不能读取CUDA tensor 需要将它转化为 CPU tensor

解决

number = dataList[dataList_size - 2].detach().numpy()

变为

number = dataList[dataList_size - 2].detach().cpu().numpy()

标签:常见问题,tensor,汇总,validity,label,cuda,深度,GPU,numpy
From: https://www.cnblogs.com/shixuanliu/p/16991330.html

相关文章