Skip to content

Commit 75cdd79

Browse files
.github/workflows/windows-trt.yml: update to tensorrt 11.1.0 and cuda 13.3.0
1 parent 1a14847 commit 75cdd79

2 files changed

Lines changed: 13 additions & 7 deletions

File tree

.github/workflows/windows-trt.yml

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -46,17 +46,17 @@ jobs:
4646
uses: actions/cache@v5
4747
with:
4848
path: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
49-
key: ${{ runner.os }}-vstrt-cuda-13.2.1
49+
key: ${{ runner.os }}-vstrt-cuda-13.3.0
5050

5151
- name: Setup CUDA
5252
if: steps.cache-cuda.outputs.cache-hit != 'true'
5353
run: |
54-
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/13.2.1/network_installers/cuda_13.2.1_windows_network.exe
55-
cuda_installer.exe -s nvcc_13.2 cudart_13.2 cuda_profiler_api_13.2 crt_13.2 nvptxcompiler_13.2
54+
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/13.3.0/network_installers/cuda_13.3.0_windows_network.exe
55+
cuda_installer.exe -s nvcc_13.3 cudart_13.3 cuda_profiler_api_13.3 crt_13.3 nvptxcompiler_13.3
5656
5757
- name: Download TensorRT
5858
run: |
59-
curl -L -o trt.zip https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.0.0/zip/TensorRT-Enterprise-11.0.0.114-Windows-amd64-cuda-13.2-Release-external.zip
59+
curl -L -o trt.zip https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/zip/TensorRT-Enterprise-11.1.0.106-Windows-amd64-cuda-13.3-Release-external.zip
6060
unzip trt.zip
6161
mv TensorRT-*/ tensorrt/
6262
@@ -71,7 +71,7 @@ jobs:
7171
-D CMAKE_BUILD_TYPE=Release
7272
-D CMAKE_CXX_FLAGS="/EHsc /Wall /wd4100 /wd4625 /wd4626 /wd4710 /wd4711 /wd4820 /wd4996 /wd5026 /wd5027"
7373
-D CMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded
74-
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v13.2"
74+
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v13.3"
7575
-D VAPOURSYNTH_INCLUDE_DIRECTORY="%cd%\vapoursynth\include"
7676
-D TENSORRT_HOME="%cd%\tensorrt"
7777
-D USE_NVINFER_PLUGIN=ON
@@ -87,7 +87,7 @@ jobs:
8787
uses: actions/checkout@v6
8888
with:
8989
repository: NVIDIA/TensorRT
90-
ref: v11.0
90+
ref: v11.1
9191
fetch-depth: 1
9292
path: tensorrt-oss
9393

@@ -108,9 +108,10 @@ jobs:
108108
run: cmake -S ../tensorrt-oss/samples/trtexec -B build_trtexec -G Ninja
109109
-D CMAKE_BUILD_TYPE=Release
110110
-D CMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded
111-
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v13.2"
111+
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v13.3"
112112
-D CMAKE_UNITY_BUILD=ON -D CMAKE_UNITY_BUILD_BATCH_SIZE=0
113113
-D CMAKE_CXX_STANDARD=20
114+
-D CMAKE_MODULE_PATH="D:/a/tensorrt-osscmake/modules"
114115

115116
- name: Build trtexec
116117
run: cmake --build build_trtexec --verbose

vstrt/trtexec/CMakeLists.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,10 @@ project(trtexec LANGUAGES CXX)
44

55
find_package(CUDAToolkit REQUIRED)
66

7+
if(NOT NLOHMANN_JSON_INCLUDE_DIRS)
8+
include(FetchNlohmannJson)
9+
endif()
10+
711
add_executable(trtexec
812
$<$<PLATFORM_ID:Windows>: longpath.manifest>
913
trtexec.cpp
@@ -27,6 +31,7 @@ target_include_directories(trtexec PRIVATE
2731
..
2832
../../include
2933
../../shared
34+
${NLOHMANN_JSON_INCLUDE_DIRS}
3035
)
3136

3237
target_link_libraries(trtexec PRIVATE CUDA::cudart_static)

0 commit comments

Comments
 (0)