查看: 1815|回复: 1

libtorch耗时问题

[复制链接]

3

主题

4

帖子

39

积分

办事员

Rank: 1

积分
39
发表于 2021-6-7 11:24:27 | 显示全部楼层 |阅读模式
在EVO xavier平台,使用libtorch做网络推理任务时候,发现了一个异常现象,下面这句话耗时不正常
  1. torch::Tensor post_center_range = torch::tensor({-70.0, -70.0, -6.0, 70.0, 70.0, 0.0}, torch::TensorOptions().dtype(torch::kFloat32).device(torch::kCUDA));
复制代码


这一句话竟然耗时82ms。

测试时候,GPU占用约90%

请问这是什么原因呢?
回复

使用道具 举报

0

主题

134

帖子

284

积分

科长

Rank: 3Rank: 3

积分
284
发表于 2021-6-7 12:50:22 | 显示全部楼层
您好

cuda性能相关的问题,建议您去以下NV论坛提问,官方的回复会更加专业。
https://forums.developer.nvidia. ... on-embedded-systems
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 点我注册

本版积分规则


快速回复 返回顶部 返回列表