Skip to content

Commit

Permalink
Windows build chang
Browse files Browse the repository at this point in the history
  • Loading branch information
mc-nv committed Dec 8, 2022
1 parent ee02cd7 commit af5130e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions Dockerfile.win10.min
Original file line number Diff line number Diff line change
Expand Up @@ -106,18 +106,18 @@ ARG CUDNN_ZIP=cudnn-windows-x86_64-${CUDNN_VERSION}_cuda11-archive.zip
LABEL CUDNN_VERSION="${CUDNN_VERSION}"

ADD https://developer.download.nvidia.com/compute/cuda/${CUDA_VERSION}/network_installers/cuda_${CUDA_VERSION}_windows_network.exe cuda_${CUDA_VERSION}_windows_network.exe
COPY ${CUDNN_ZIP} .
COPY ${TENSORRT_ZIP} .
COPY ${CUDNN_ZIP} /tmp/${CUDNN_ZIP}
COPY ${TENSORRT_ZIP} /tmp/${TENSORRT_ZIP}

LABEL CUDA_VERSION="${CUDA_VERSION}"

RUN unzip %TENSORRT_ZIP%
RUN unzip /tmp/%TENSORRT_ZIP%
RUN move TensorRT-* TensorRT
ENV TRT_VERSION ${TENSORRT_VERSION}

RUN cuda_%CUDA_VERSION%_windows_network.exe -s %CUDA_PACKAGES%

RUN unzip %CUDNN_ZIP%
RUN unzip /tmp/%CUDNN_ZIP%
RUN move cudnn-* cudnn
RUN copy cudnn\bin\cudnn*.dll "%CUDA_INSTALL_ROOT_WP%\bin\."
RUN copy cudnn\lib\x64\cudnn*.lib "%CUDA_INSTALL_ROOT_WP%\lib\x64\."
Expand Down

0 comments on commit af5130e

Please sign in to comment.