diff --git a/win/pkgs/tensorrt.ps1 b/win/pkgs/tensorrt.ps1 index ba53cedd..cb013e66 100755 --- a/win/pkgs/tensorrt.ps1 +++ b/win/pkgs/tensorrt.ps1 @@ -27,7 +27,7 @@ pushd "$root" # Update the URL as new version releases. $trt_mirror="https://github.com/xkszltl/Roaster/releases/download/trt" -$trt_name="TensorRT-7.1.3.4.Windows10.x86_64.cuda-11.0.cudnn8.0.zip" +$trt_name="TensorRT-7.2.1.6.Windows10.x86_64.cuda-$((nvcc --version) -match ' release ([0-9\.]*)' -replace '.* release ([0-9\.]*).*','${1}').cudnn8.0.zip" if (-not (Test-Path "../${trt_name}")) {