Skip to content

Commit 304e64a

Browse files
authored
Pin TRT version (#103)
Pin TensorRT Version in Stable Diffusion Tutorial
1 parent a4bd15b commit 304e64a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Popular_Models_Guide/StableDiffusion/docker/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ ARG BASE_IMAGE_TAG=24.01-py3
2929

3030
FROM ${BASE_IMAGE}:${BASE_IMAGE_TAG} as tritonserver-stable-diffusion
3131

32-
RUN pip install --pre --upgrade --extra-index-url https://pypi.nvidia.com tensorrt
32+
RUN pip install --pre --upgrade --extra-index-url https://pypi.nvidia.com tensorrt==9.2.0.post12.dev5
3333

3434
RUN git clone https://github.com/NVIDIA/TensorRT.git -b release/9.2 --single-branch /tmp/TensorRT
3535

0 commit comments

Comments
 (0)