From e50dd6ab20b4027c9799b055c1b75754bf0028ea Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Mon, 16 Sep 2024 15:03:26 -0600 Subject: [PATCH] Update comment --- docker/tensorrt/Dockerfile.amd64 | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/tensorrt/Dockerfile.amd64 b/docker/tensorrt/Dockerfile.amd64 index 343e6979d..6022b4e50 100644 --- a/docker/tensorrt/Dockerfile.amd64 +++ b/docker/tensorrt/Dockerfile.amd64 @@ -12,7 +12,7 @@ ARG TARGETARCH COPY docker/tensorrt/requirements-amd64.txt /requirements-tensorrt.txt RUN mkdir -p /trt-wheels && pip3 wheel --wheel-dir=/trt-wheels -r /requirements-tensorrt.txt -# Build TensorRT-specific library +# Build CuDNN FROM tensorrt-base AS cudnn-deps ARG COMPUTE_LEVEL