Skip to content

Commit ed86133

Browse files
author
deadeyegoodwin
committed
Fix naming of windows cudnn package (#3861)
1 parent 4f925a7 commit ed86133

File tree

2 files changed

+14
-11
lines changed

2 files changed

+14
-11
lines changed

Dockerfile.win10.min

+5-4
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,8 @@ ARG TENSORRT_VERSION=8.2.2.1
100100
ARG TENSORRT_ZIP=TensorRT-${TENSORRT_VERSION}.Windows10.x86_64.cuda-11.4.cudnn8.2.zip
101101

102102
ARG CUDNN_VERSION=8.3.2.44
103-
ARG CUDNN_ZIP=cudnn-11.5-windows-x64-v${CUDNN_VERSION}.zip
103+
ARG CUDNN_DIR=cudnn-windows-x86_64-${CUDNN_VERSION}_cuda11.5-archive
104+
ARG CUDNN_ZIP=${CUDNN_DIR}.zip
104105

105106
WORKDIR /tmp
106107
ADD https://developer.download.nvidia.com/compute/cuda/${CUDA_VERSION}/network_installers/cuda_${CUDA_VERSION}_win10_network.exe cuda_${CUDA_VERSION}_win10_network.exe
@@ -115,9 +116,9 @@ ENV TRT_VERSION ${TENSORRT_VERSION}
115116
WORKDIR /tmp
116117
RUN cuda_%CUDA_VERSION%_win10_network.exe -s %CUDA_PACKAGES%
117118
RUN unzip %CUDNN_ZIP%
118-
RUN copy libcudnn\bin\cudnn*.dll "%CUDA_INSTALL_ROOT_WP%\bin\."
119-
RUN copy libcudnn_dev\lib\x64\cudnn*.lib "%CUDA_INSTALL_ROOT_WP%\lib\x64\."
120-
RUN copy libcudnn_dev\include\cudnn*.h "%CUDA_INSTALL_ROOT_WP%\include\."
119+
RUN copy %CUDNN_DIR%\bin\cudnn*.dll "%CUDA_INSTALL_ROOT_WP%\bin\."
120+
RUN copy %CUDNN_DIR%\lib\cudnn*.lib "%CUDA_INSTALL_ROOT_WP%\lib\x64\."
121+
RUN copy %CUDNN_DIR%\include\cudnn*.h "%CUDA_INSTALL_ROOT_WP%\include\."
121122

122123
# Copy the CUDA visualstudio integration from where is was installed
123124
# into the appropriate place in BuildTools

docs/build.md

+9-7
Original file line numberDiff line numberDiff line change
@@ -247,14 +247,16 @@ the Windows build. The Windows min container is
247247
[Dockerfile.win10.min](../Dockerfile.win10.min).
248248

249249
Before building the min container you must download the appropriate
250-
cuDNN and TensorRT versions and place them in the local directory.
250+
cuDNN and TensorRT versions and place them in the same directory as
251+
Dockerfile.win10.min.
251252

252-
* For cuDNN the CUDNN_VERSION and CUDNN_ZIP arguments indicate the
253-
version of cuDNN that your should download from
254-
https://developer.nvidia.com/rdp/cudnn-download.
253+
* For cuDNN the CUDNN_VERSION and CUDNN_ZIP arguments defined in
254+
Dockerfile.win10.min indicate the version of cuDNN that your should
255+
download from https://developer.nvidia.com/rdp/cudnn-download.
255256

256-
* For TensorRT the TENSORRT_VERSION and TENSORRT_ZIP arguments
257-
indicate the version of TensorRT that your should download from
257+
* For TensorRT the TENSORRT_VERSION and TENSORRT_ZIP arguments defined
258+
in Dockerfile.win10.min indicate the version of TensorRT that your
259+
should download from
258260
https://developer.nvidia.com/nvidia-tensorrt-download.
259261

260262
After downloading the zip files for cuDNN and TensorRT, you build the
@@ -274,7 +276,7 @@ and so you must enable them explicitly. The following build.py
274276
invocation builds all features and backends available on windows.
275277

276278
```bash
277-
python build.py --cmake-dir=<path/to/repo>/build --build-dir=/tmp/citritonbuild --no-container-pull --image=base,win10-py3-min --enable-logging --enable-stats --enable-tracing --enable-gpu --endpoint=grpc --endpoint=http --repo-tag=common:<container tag> --repo-tag=core:<container tag> --repo-tag=backend:<container tag> --repo-tag=thirdparty:<container tag> --backend=ensemble --backend=tensorrt:<container tag> --backend=onnxruntime:<container tag>
279+
python build.py --cmake-dir=<path/to/repo>/build --build-dir=/tmp/citritonbuild --no-container-pull --image=base,win10-py3-min --enable-logging --enable-stats --enable-tracing --enable-gpu --endpoint=grpc --endpoint=http --repo-tag=common:<container tag> --repo-tag=core:<container tag> --repo-tag=backend:<container tag> --repo-tag=thirdparty:<container tag> --backend=ensemble --backend=tensorrt:<container tag> --backend=onnxruntime:<container tag> --backend=openvino:<container tag>
278280
```
279281

280282
If you are building on *main* branch then '<container tag>' will

0 commit comments

Comments
 (0)