DeepStream社区
Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
墨衍会员 · AI 创作全网分发 墨衍智能分发 通过墨衍一键同步至各平台 可同步平台 CSDN微信公众号微博知乎掘金百家号博客园抖音小红书 你的文章也可以这样分发 墨衍会员 ¥399起/年，写一次发全网
我也要 关注 + 加入社区输入的是gpu的torch，但是权重weight是cpu的，所以部匹配报错
重新到处模型使用gpu导出：
python export.py --weights runs/train/exp/weights/best.pt --include torchscript --device 0
...全文
相关推荐
pytorch 查看cuda 版本方式
例如，下面的代码是错误的：```pythonself.W1 = nn.Parameter(torch.FloatTensor(3, 3), requires_grad=True).cuda()```如描述中提到
pytorch cuda上tensor的定义 以及减少cpu的操作详解
例如，可以在GPU上直接定义张量：```pythontorch.zeros().cuda()```或者，对于特定形状的张量，可以使用：```pythontorch.cuda.FloatTensor(batch_size
pytorch.docx
- **特殊函数**：`torch.rsqrt(input)`、`torch.lerp(start,end,weight)` 和 `torch.tanh(input,out=None)` 可以分别计算张量的倒数平方根
PyTorch中张量的创建方法
如果要在GPU上进行运算，可以使用对应的GPU张量类型，如`torch.cuda.FloatTensor`。张量的形状可以通过切片、堆叠、展平等操作进行改变。
VDA 19.1_3rd edition_2026_English.rar
VDA 19.1_3rd edition_2026_English.rar
