torch2trt icon indicating copy to clipboard operation
torch2trt copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA-AI-IOT

GPU Device specify

Open TaoKai opened this issue 1 year ago • 1 comments

hello friend, how can I specify the GPU device number for a TRTModule? such as cuda:0 or cuda:1 etc

TaoKai avatar Jun 02 '23 07:06 TaoKai

i have same problem with you..

bi3qwq avatar Jul 23 '23 03:07 bi3qwq