Skip to content

Commit

Permalink
.github/workflows/windows-trt.yml: upgrade to trt 10
Browse files Browse the repository at this point in the history
  • Loading branch information
WolframRhodium committed Mar 27, 2024
1 parent 50478c0 commit 48d8946
Showing 1 changed file with 12 additions and 25 deletions.
37 changes: 12 additions & 25 deletions .github/workflows/windows-trt.yml
Original file line number Diff line number Diff line change
Expand Up @@ -46,31 +46,20 @@ jobs:
uses: actions/cache@v4
with:
path: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
key: ${{ runner.os }}-vstrt-cuda-12.3.1
key: ${{ runner.os }}-vstrt-cuda-12.4.0
save-always: true

- 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.3.1/network_installers/cuda_12.3.1_windows_network.exe
cuda_installer.exe -s nvcc_12.3 cudart_12.3 nvprof_12.3 cuda_profiler_api_12.3 thrust_12.3 cublas_dev_12.3
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/12.4.0/network_installers/cuda_12.4.0_windows_network.exe
cuda_installer.exe -s nvcc_12.4 cudart_12.4
- name: Checkout tensorrt
uses: actions/checkout@v4
with:
repository: AmusementClub/cuda
token: ${{ secrets.REPO_TOKEN }}
ref: tensorrt-9.2.0
path: tensorrt

- name: Download cuDNN
- name: Download TensorRT
run: |
curl -LJ https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/cudnn-windows-x86_64-8.9.6.50_cuda12-archive.zip -o cudnn.zip
unzip cudnn.zip
mkdir -p cudnn
mv cudnn-windows-*/include -t cudnn
mv cudnn-windows-*/lib -t cudnn
ls -R cudnn
curl -L -o trt.zip https://developer.download.nvidia.com/compute/machine-learning/tensorrt/10.0.0/zip/TensorRT-10.0.0.6.Windows10.win10.cuda-12.4.zip
unzip trt.zip
move "TensorRT-*" tensorrt
- name: Download VapourSynth headers
run: |
Expand All @@ -82,11 +71,10 @@ 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.3"
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4"
-D VAPOURSYNTH_INCLUDE_DIRECTORY="%cd%\vapoursynth\include"
-D TENSORRT_HOME="%cd%\..\tensorrt\tensorrt"
-D TENSORRT_HOME="%cd%\tensorrt"
-D USE_NVINFER_PLUGIN=ON
-D CUDNN_HOME="cudnn"

- name: Build
run: cmake --build build --config Release --verbose
Expand All @@ -98,7 +86,7 @@ jobs:
uses: actions/checkout@v4
with:
repository: NVIDIA/TensorRT
ref: release/9.2
ref: release/9.3
fetch-depth: 1
path: tensorrt-oss

Expand All @@ -111,9 +99,8 @@ 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.3"
-D TENSORRT_HOME="%cd%\..\tensorrt\tensorrt"
-D CUDNN_HOME="cudnn"
-D CUDAToolkit_ROOT="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4"
-D TENSORRT_HOME="%cd%\tensorrt"

- name: Build trtexec
run: cmake --build build_trtexec --verbose
Expand Down

0 comments on commit 48d8946

Please sign in to comment.