问题确认 Search before asking
- 我已经查询历史issue,没有发现类似的bug。I have searched the issues and found no similar bug report.
To pick up a draggable item, press the space bar.
While dragging, use the arrow keys to move the item.
Press space again to drop the item in its new position, or press escape to cancel.
Bug组件 Bug Component
No response
Bug描述 Describe the Bug
我想使用两个一样的gpu 同时推理报错。单独使用gpu 0,gpu 1时候都能运行。同时初始化两个gpu 的模型没有报错,执行推理时候报错了:CUDA Failure at core.cpp:218: invalid argument
复现环境 Environment
Windows
Version 6.3
tensorrt 10×
Cuda 12.8
Bug描述确认 Bug description confirmation
- 我确认已经提供了Bug复现步骤、代码改动说明、以及环境信息,确认问题是可以复现的。I confirm that the bug replication steps, code change instructions, and environment information have been provided, and the problem can be reproduced.
To pick up a draggable item, press the space bar.
While dragging, use the arrow keys to move the item.
Press space again to drop the item in its new position, or press escape to cancel.
© 版权声明
文章版权归作者所有,未经允许请勿转载。如内容涉嫌侵权,请在本页底部进入<联系我们>进行举报投诉!
THE END
- 最新
- 最热
只看作者测试代码使用example里面的segment,只是在里面添加了option.setDevice设置GPU
@zw1765 应该是每次切换设备都要
cudaSetDevice
,但目前只在初始化时这么做了。你可以先自己改一下,在
Trtbackend::infer
也加上
cudaSetDevice
,试一试是否可行。