快捷导航
1 5125

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

精彩推荐

  • 盒子重启之后打不开了,启动界面一直跳动
  • APEX AD 10远程控制问题
  • apex ad 10 重新装系统后连接热点死机,连
  • ros2相机功能包编译失败

明星用户