快捷导航
1 4546

libtorch耗时问题

hahahhahah 于 2021-6-7 11:24 发表 [复制链接]
在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%

请问这是什么原因呢?
举报 使用道具
| 回复

共 1 个关于本帖的回复 最后回复于 2021-6-7 12:50

贺老师 游客 发表于 2021-6-7 12:50 | 显示全部楼层
您好

cuda性能相关的问题,建议您去以下NV论坛提问,官方的回复会更加专业。
https://forums.developer.nvidia. ... on-embedded-systems
您需要登录后才可以回帖 登录 | 点我注册

精彩推荐

  • canbus与vcu相连接,出现bus-off状态
  • 有线连接失败
  • Apex 串口通讯
  • 关于SPI通信问题咨询
  • MIIVII APEX DUAL ORIN米文域控制器产品合

明星用户