StreamDiffusion icon indicating copy to clipboard operation
StreamDiffusion copied to clipboard

Use tensorrt docker image instead of installing tensorrt in dockerfile build

Open henryclw opened this issue 1 year ago • 0 comments

According to Nvidia's documents, there is a docker image tagged as nvcr.io/nvidia/tensorrt:23.01-py3, could we use this as base image instead of using nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 as base image? https://github.com/cumulo-autumn/StreamDiffusion/blob/62596fd39b86a3088e28b2f2c58caa36bc069c50/Dockerfile#L1 https://docs.nvidia.com/deeplearning/tensorrt/container-release-notes/running.html

henryclw avatar Dec 26 '23 19:12 henryclw