diff --git a/.github/workflows/windows-trt.yml b/.github/workflows/windows-trt.yml index 783b298..f0a8f04 100644 --- a/.github/workflows/windows-trt.yml +++ b/.github/workflows/windows-trt.yml @@ -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/