Building docker image

Question

Building docker image

Closed this issue a year ago · 1 comments

Hello,

Thank you for sharing your repository and details on running your code. I've been having issues building the docker image where the final Torch2trt fails due to missing the CUDA_HOME variable. I've tried the install process without Torch2trt but found that Cuda is unavailable. Do you have any suggestions on how to remedy this issue?

Best,

Answer 1 · 2023-09-28T10:51:20.000Z

Hi, was CUDA available on your system, e.g nvcc -V
You can skip Torch2Trt installation if you want, and also no TensorRT at all. But you need to modify some code to make it work with Torch model instead of TensorRT, especially the YOLOX related-parts.