Skip to content

Commit

Permalink
.github/workflows/windows-trt.yml: update to cuda 12.6.3 and tensorrt…
Browse files Browse the repository at this point in the history
… 10.7.0
  • Loading branch information
WolframRhodium committed Dec 3, 2024
1 parent ac25053 commit 9a7dfac
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.2
key: ${{ runner.os }}-vstrt-cuda-12.6.3
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.2/network_installers/cuda_12.6.2_windows_network.exe
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/12.6.3/network_installers/cuda_12.6.3_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.6.0/zip/TensorRT-10.6.0.26.Windows.win10.cuda-12.6.zip
curl -L -o trt.zip https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.7.0/zip/TensorRT-10.7.0.23.Windows.win10.cuda-12.6.zip
unzip trt.zip
mv TensorRT-*/ tensorrt/
Expand Down

0 comments on commit 9a7dfac

Please sign in to comment.