I have trained four different tensorrt models and the machine have four gpus, how can I allocate different tensorrt model to the different gpu, it looks like all the trt engine were allocated in the gpu 0 by default. But when I use “cudaSetDevice” in my code(I have to run all the models in one process and the code is C++ version), the message “WARNING: Using an engine plan file across different models of devices is not recommended and is likely to affect performance or even cause errors.” was reported, and all the models can’t run normally. Can you give me an solution? Thanks a lot!
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| How to run TensorRT on a muliti-GPU platform. | 1 | 3554 | November 27, 2018 | |
| How to inference with tensorrt on multi gpus in python | 2 | 2317 | April 9, 2021 | |
| [TRT7 Python API] How to assign a TRT engine to a specific GPU? | 2 | 1322 | July 18, 2020 | |
| Concurrently run two or more engine on a tensorrt | 3 | 2543 | April 14, 2020 | |
| Bug : Tensorrt Model not loading on same GPU on a different device (slight driver version difference) | 1 | 357 | April 30, 2024 | |
| Run multiple model(engine) with tensorrt without deepstream | 1 | 1224 | April 20, 2020 | |
| Assign device for TensorRT inference API | 2 | 587 | June 2, 2020 | |
| How can we build a tensorRT model just once and run on different GPUs? | 3 | 566 | May 5, 2020 | |
| TensorRT MultiThread with MultiGPU | 1 | 631 | February 14, 2023 | |
| Using Multiple GPUS | 0 | 606 | August 20, 2019 |