Skip to content

Commit

Permalink
.github/workflows/windows-trt.yml: update to cuda 12.6.1 and tensorrt…
Browse files Browse the repository at this point in the history
… 10.5.0
  • Loading branch information
WolframRhodium committed Oct 1, 2024
1 parent 0b7213d commit c4d80f8
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/windows-trt.yml
Original file line number Diff line number Diff line change
Expand Up @@ -46,18 +46,18 @@ jobs:
uses: actions/cache@v4
with:
path: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
key: ${{ runner.os }}-vstrt-cuda-12.6.0
key: ${{ runner.os }}-vstrt-cuda-12.6.1
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.6.0/network_installers/cuda_12.6.0_windows_network.exe
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/12.6.1/network_installers/cuda_12.6.1_windows_network.exe
cuda_installer.exe -s nvcc_12.6 cudart_12.6 cuda_profiler_api_12.6
- name: Download TensorRT
run: |
curl -L -o trt.zip https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.4.0/zip/TensorRT-10.4.0.26.Windows.win10.cuda-12.6.zip
curl -L -o trt.zip https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.5.0/zip/TensorRT-10.5.0.18.Windows.win10.cuda-12.6.zip
unzip trt.zip
mv TensorRT-*/ tensorrt/
Expand Down

0 comments on commit c4d80f8

Please sign in to comment.