Skip to content

Commit

Permalink
.github/workflows/windows-trt.yml: update to TensorRT 9.1.0
Browse files Browse the repository at this point in the history
  • Loading branch information
WolframRhodium committed Oct 22, 2023
1 parent daf9620 commit 25f3fe2
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions .github/workflows/windows-trt.yml
Original file line number Diff line number Diff line change
Expand Up @@ -46,20 +46,20 @@ jobs:
uses: actions/cache@v3
with:
path: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
key: ${{ runner.os }}-vstrt-cuda-12.1.1
key: ${{ runner.os }}-vstrt-cuda-12.2.2

- name: Setup CUDA
if: steps.cache-cuda.outputs.cache-hit != 'true'
run: |
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/12.1.1/network_installers/cuda_12.1.1_windows_network.exe
cuda_installer.exe -s nvcc_12.1 cudart_12.1 nvprof_12.1 cuda_profiler_api_12.1 thrust_12.1
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/12.2.2/network_installers/cuda_12.2.2_windows_network.exe
cuda_installer.exe -s nvcc_12.2 cudart_12.2 nvprof_12.2 cuda_profiler_api_12.2 thrust_12.2
- name: Checkout tensorrt
uses: actions/checkout@v3
with:
repository: AmusementClub/cuda
token: ${{ secrets.REPO_TOKEN }}
ref: tensorrt-8.6.1
ref: tensorrt-9.1.0
path: tensorrt

- name: Download VapourSynth headers
Expand All @@ -72,7 +72,7 @@ jobs:
run: cmake -S . -B build -G Ninja -LA
-D CMAKE_BUILD_TYPE=Release
-D CMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1"
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2"
-D VAPOURSYNTH_INCLUDE_DIRECTORY="%cd%\vapoursynth\include"
-D TENSORRT_HOME="%cd%\..\tensorrt\tensorrt"
-D USE_NVINFER_PLUGIN=ON
Expand All @@ -87,7 +87,7 @@ jobs:
uses: actions/checkout@v3
with:
repository: NVIDIA/TensorRT
ref: v8.6.1
ref: v9.1.0
fetch-depth: 1
path: tensorrt-oss

Expand All @@ -100,7 +100,7 @@ jobs:
run: cmake -S ../tensorrt-oss/samples/trtexec -B build_trtexec -G Ninja
-D CMAKE_BUILD_TYPE=Release
-D CMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1"
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2"
-D TENSORRT_HOME="%cd%\..\tensorrt\tensorrt"

- name: Build trtexec
Expand Down

0 comments on commit 25f3fe2

Please sign in to comment.