Model conversion on Multi-GPU computer

[Zhong-zh13]

Hello and thank you for your wonderful tool. It went so well! I just have a small question.

Now I have 4 GPUs on my computer. Since TensorRT is more of a device specific thing, how could we select the GPU device to use when running the c++ program for ONNX->TensorRT conversion?

I tried to search in the codes but failed to find the correct part.

Thank you very much

[linghu8812]

a simple way: use CUDA_VISIBLE_DEVICE

[Zhong-zh13]

@linghu8812 Thank you for your reply! I've already known how to modify the c++ code to achieve that. 再次感谢您的工具