Update comment

This commit is contained in:
Nicolas Mowen 2024-09-16 15:03:26 -06:00
parent feee4a52a3
commit e50dd6ab20

View File

@ -12,7 +12,7 @@ ARG TARGETARCH
COPY docker/tensorrt/requirements-amd64.txt /requirements-tensorrt.txt COPY docker/tensorrt/requirements-amd64.txt /requirements-tensorrt.txt
RUN mkdir -p /trt-wheels && pip3 wheel --wheel-dir=/trt-wheels -r /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 FROM tensorrt-base AS cudnn-deps
ARG COMPUTE_LEVEL ARG COMPUTE_LEVEL