Upgrade TensorRT to 7.2.1 on Windows and use nvcc to figure out version.

xkszltl · Oct 31, 2020 · 1a11fd0 · 1a11fd0
1 parent ce13a47
commit 1a11fd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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}"))
 {