StreamDiffusion
StreamDiffusion copied to clipboard
Use tensorrt docker image instead of installing tensorrt in dockerfile build
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