@@ -37,9 +37,9 @@ RUN choco install unzip -y
3737#
3838# Installing TensorRT
3939#
40- ARG TENSORRT_VERSION
41- ARG TENSORRT_ZIP="TensorRT-${TENSORRT_VERSION}.Windows10.x86_64.cuda-12.0 .zip"
42- ARG TENSORRT_SOURCE=${TENSORRT_ZIP}
40+ ARG TENSORRT_VERSION=10.0.1.6
41+ ARG TENSORRT_ZIP="TensorRT-${TENSORRT_VERSION}.Windows10.x86_64.cuda-12.4 .zip"
42+ ARG TENSORRT_SOURCE=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.0.1/zip/TensorRT-10.0.1.6.Windows10.win10.cuda-12.4.zip
4343# COPY ${TENSORRT_ZIP} /tmp/${TENSORRT_ZIP}
4444ADD ${TENSORRT_SOURCE} /tmp/${TENSORRT_ZIP}
4545RUN unzip /tmp/%TENSORRT_ZIP%
@@ -51,9 +51,9 @@ LABEL TENSORRT_VERSION="${TENSORRT_VERSION}"
5151#
5252# Installing cuDNN
5353#
54- ARG CUDNN_VERSION
54+ ARG CUDNN_VERSION=9.1.0.70
5555ARG CUDNN_ZIP=cudnn-windows-x86_64-${CUDNN_VERSION}_cuda12-archive.zip
56- ARG CUDNN_SOURCE=${CUDNN_ZIP}
56+ ARG CUDNN_SOURCE=https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/cudnn-windows-x86_64-9.1.0.70_cuda12-archive.zip
5757ADD ${CUDNN_SOURCE} /tmp/${CUDNN_ZIP}
5858RUN unzip /tmp/%CUDNN_ZIP%
5959RUN move cudnn-* cudnn
@@ -75,7 +75,7 @@ RUN choco install git docker unzip -y
7575#
7676# Installing python
7777#
78- ARG PYTHON_VERSION=3.8.10
78+ ARG PYTHON_VERSION=3.10.11
7979ARG PYTHON_SOURCE=https://www.python.org/ftp/python/${PYTHON_VERSION}/python-${PYTHON_VERSION}-amd64.exe
8080ADD ${PYTHON_SOURCE} python-${PYTHON_VERSION}-amd64.exe
8181RUN python-%PYTHON_VERSION%-amd64.exe /quiet InstallAllUsers=1 PrependPath=1 Include_doc=0 TargetDir="C:\python%PYTHON_VERSION%"
@@ -88,14 +88,8 @@ LABEL PYTHON_VERSION=${PYTHON_VERSION}
8888#
8989# Installing CMake
9090#
91- ARG CMAKE_VERSION=3.27.1
92- ARG CMAKE_FILE=cmake-${CMAKE_VERSION}-windows-x86_64
93- ARG CMAKE_SOURCE=https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/${CMAKE_FILE}.zip
94-
95- ADD ${CMAKE_SOURCE} ${CMAKE_FILE}.zip
96- RUN unzip %CMAKE_FILE%.zip
97- RUN move %CMAKE_FILE% "c:\CMake"
98- RUN setx PATH "c:\CMake\bin;%PATH%"
91+ ARG CMAKE_VERSION=3.29.3
92+ RUN pip install cmake==%CMAKE_VERSION%
9993
10094ENV CMAKE_TOOLCHAIN_FILE /vcpkg/scripts/buildsystems/vcpkg.cmake
10195ENV VCPKG_TARGET_TRIPLET x64-windows
@@ -104,27 +98,22 @@ LABEL CMAKE_VERSION=${CMAKE_VERSION}
10498
10599# Be aware that pip can interact badly with VS cmd shell so need to pip install before
106100# vsdevcmd.bat (see https://bugs.python.org/issue38989)
107-
108-
109101#
110102# Installing Visual Studio BuildTools: VS17 2022
111103#
112- ARG BUILDTOOLS_VERSION
104+ ARG BUILDTOOLS_VERSION=17.9.34622.214
113105# Download collect.exe in case of an install failure.
114106ADD https://aka.ms/vscollect.exe "C:\tmp\collect.exe"
115107
116108# Use the latest release channel. For more control, specify the location of an internal layout.
117- ARG CHANNEL_URL=https://aka.ms/vs/17/release/channel
118- ADD ${CHANNEL_URL} "C:\tmp\VisualStudio.chman"
119109# Download the Build Tools bootstrapper.
120- ARG BUILD_TOOLS_SOURCE=https://aka.ms/vs/17/release/vs_buildtools.exe
110+ # ARG BUILD_TOOLS_SOURCE=https://aka.ms/vs/17/release/vs_buildtools.exe
111+ ARG BUILD_TOOLS_SOURCE=https://download.visualstudio.microsoft.com/download/pr/5e7b923b-7d89-4e14-95b8-a84ab168e243/96b21d216c7954aaf606c6d7ba59a3de991884a8a86c578c767ba349c23188a9/vs_BuildTools.exe
121112ADD ${BUILD_TOOLS_SOURCE} vs_buildtools.exe
122113# Install Build Tools with the Microsoft.VisualStudio.Workload.VCTools workload, including recommended.
123114ARG VS_INSTALL_PATH_WP="C:\BuildTools"
124115RUN vs_buildtools.exe --quiet --wait --norestart --nocache install \
125116 --installPath %VS_INSTALL_PATH_WP% \
126- --channelUri "C:\tmp\VisualStudio.chman" \
127- --installChannelUri "C:\tmp\VisualStudio.chman" \
128117 --add Microsoft.VisualStudio.Workload.VCTools \
129118 --includeRecommended \
130119 --locale "En-us"
@@ -136,15 +125,15 @@ WORKDIR /
136125#
137126# Installing Vcpkg
138127#
139- ARG VCPGK_VERSION=2023.11.20
128+ ARG VCPGK_VERSION=2024.03.19
140129RUN git clone --single-branch --depth=1 -b %VCPGK_VERSION% https://github.com/microsoft/vcpkg.git
141130WORKDIR /vcpkg
142131RUN bootstrap-vcpkg.bat
143132RUN vcpkg.exe update
144133RUN vcpkg.exe install \
145- b64:x64-windows \
146134 boost-interprocess:x64-windows \
147135 boost-stacktrace:x64-windows \
136+ b64:x64-windows \
148137 openssl-windows:x64-windows \
149138 openssl:x64-windows \
150139 pthread:x64-windows \
@@ -160,8 +149,8 @@ WORKDIR /
160149# Installing CUDA
161150#
162151ARG CUDA_MAJOR=12
163- ARG CUDA_MINOR=3
164- ARG CUDA_PATCH=2
152+ ARG CUDA_MINOR=5
153+ ARG CUDA_PATCH=0
165154ARG CUDA_VERSION=${CUDA_MAJOR}.${CUDA_MINOR}.${CUDA_PATCH}
166155ARG CUDA_PACKAGES="nvcc_${CUDA_MAJOR}.${CUDA_MINOR} \
167156 cudart_${CUDA_MAJOR}.${CUDA_MINOR} \
@@ -186,15 +175,15 @@ RUN copy "%CUDA_INSTALL_ROOT_WP%\extras\visual_studio_integration\MSBuildExtensi
186175
187176RUN setx PATH "%CUDA_INSTALL_ROOT_WP%\bin;%PATH%"
188177
189- ARG CUDNN_VERSION
178+ ARG CUDNN_VERSION=9.1.0.70
190179ENV CUDNN_VERSION ${CUDNN_VERSION}
191180COPY --from=dependency_base /cudnn /cudnn
192181RUN copy cudnn\bin\cudnn*.dll "%CUDA_INSTALL_ROOT_WP%\bin\."
193182RUN copy cudnn\lib\x64\cudnn*.lib "%CUDA_INSTALL_ROOT_WP%\lib\x64\."
194183RUN copy cudnn\include\cudnn*.h "%CUDA_INSTALL_ROOT_WP%\include\."
195184LABEL CUDNN_VERSION="${CUDNN_VERSION}"
196185
197- ARG TENSORRT_VERSION
186+ ARG TENSORRT_VERSION=10.0.1.6
198187ENV TRT_VERSION ${TENSORRT_VERSION}
199188COPY --from=dependency_base /TensorRT /TensorRT
200189RUN setx PATH "c:\TensorRT\lib;%PATH%"
0 commit comments