@@ -37,9 +37,9 @@ RUN choco install unzip -y
37
37
#
38
38
# Installing TensorRT
39
39
#
40
- ARG TENSORRT_VERSION=10.3 .0.26
41
- ARG TENSORRT_ZIP="TensorRT-${TENSORRT_VERSION}.Windows10.x86_64 .cuda-12.5 .zip"
42
- ARG TENSORRT_SOURCE=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.2 .0/zip/TensorRT-10.2 .0.19.Windows10.x86_64 .cuda-12.5 .zip
40
+ ARG TENSORRT_VERSION=10.4 .0.26
41
+ ARG TENSORRT_ZIP="TensorRT-${TENSORRT_VERSION}.Windows.win10 .cuda-12.6 .zip"
42
+ ARG TENSORRT_SOURCE=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.4 .0/zip/TensorRT-10.4 .0.26.Windows.win10 .cuda-12.6 .zip
43
43
# COPY ${TENSORRT_ZIP} /tmp/${TENSORRT_ZIP}
44
44
ADD ${TENSORRT_SOURCE} /tmp/${TENSORRT_ZIP}
45
45
RUN unzip /tmp/%TENSORRT_ZIP%
@@ -51,9 +51,9 @@ LABEL TENSORRT_VERSION="${TENSORRT_VERSION}"
51
51
#
52
52
# Installing cuDNN
53
53
#
54
- ARG CUDNN_VERSION=9.3 .0.75
54
+ ARG CUDNN_VERSION=9.4 .0.58
55
55
ARG CUDNN_ZIP=cudnn-windows-x86_64-${CUDNN_VERSION}_cuda12-archive.zip
56
- ARG CUDNN_SOURCE=https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/cudnn-windows-x86_64-9.2.1.18_cuda12 -archive.zip
56
+ ARG CUDNN_SOURCE=https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/cudnn-windows-x86_64-9.4.0.58_cuda12 -archive.zip
57
57
ADD ${CUDNN_SOURCE} /tmp/${CUDNN_ZIP}
58
58
RUN unzip /tmp/%CUDNN_ZIP%
59
59
RUN move cudnn-* cudnn
@@ -175,15 +175,15 @@ RUN copy "%CUDA_INSTALL_ROOT_WP%\extras\visual_studio_integration\MSBuildExtensi
175
175
176
176
RUN setx PATH "%CUDA_INSTALL_ROOT_WP%\bin;%PATH%"
177
177
178
- ARG CUDNN_VERSION=9.3 .0.75
178
+ ARG CUDNN_VERSION=9.4 .0.58
179
179
ENV CUDNN_VERSION ${CUDNN_VERSION}
180
180
COPY --from=dependency_base /cudnn /cudnn
181
181
RUN copy cudnn\bin\cudnn*.dll "%CUDA_INSTALL_ROOT_WP%\bin\."
182
182
RUN copy cudnn\lib\x64\cudnn*.lib "%CUDA_INSTALL_ROOT_WP%\lib\x64\."
183
183
RUN copy cudnn\include\cudnn*.h "%CUDA_INSTALL_ROOT_WP%\include\."
184
184
LABEL CUDNN_VERSION="${CUDNN_VERSION}"
185
185
186
- ARG TENSORRT_VERSION=10.3 .0.26
186
+ ARG TENSORRT_VERSION=10.4 .0.26
187
187
ENV TRT_VERSION ${TENSORRT_VERSION}
188
188
COPY --from=dependency_base /TensorRT /TensorRT
189
189
RUN setx PATH "c:\TensorRT\lib;%PATH%"
0 commit comments