search

NVIDIA / TensorRT

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
https://developer.nvidia.com/tensorrt
Apache License 2.0
10.89k stars 2.14k forks source link

Image generation failure of TensorRT 8.6 when running demo/Diffusion/demo_txt2img.py on GPU v100 #3518

Open xieyongliang opened 1 year ago

xieyongliang commented 1 year ago

Description

Environment

TensorRT Version: 8.6

NVIDIA GPU: v100

NVIDIA Driver Version: 535.104.12

CUDA Version: 12.2

CUDNN Version:

Operating System: Ubuntu 20.04

Python Version (if applicable): Python 3.8.10

Tensorflow Version (if applicable):

PyTorch Version (if applicable): 1.13.1/2.1.1

Baremetal or Container (if so, version): nvcr.io/nvidia/pytorch:23.06-py3

Relevant Files

Model link:

Steps To Reproduce

Commands or scripts:

Have you tried the latest release?:

Can this model run on other frameworks? For example run ONNX model with ONNXRuntime (polygraphy run <model.onnx> --onnxrt):

zerollzeng commented 12 months ago

What error did you see?