快捷导航
1 4898

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
您需要登录后才可以回帖 登录 | 点我注册

精彩推荐

  • MIIVII EVO ORIN的最新镜像中米文的源报404
  • 求助,储存空间不够,如何解决
  • APEX AD 10问题见图片
  • 相机启动launch
  • APEX AD-10 GPS授时修改串口波特率

明星用户