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

xkszltl · xkszltl · commit 1a11fd0e0157 · 2020-11-01T01:25:46.000+08:00
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}"))
 {

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ pushd "$root"`
`27`	`27`
`28`	`28`	`# Update the URL as new version releases.`
`29`	`29`	`$trt_mirror="https://github.com/xkszltl/Roaster/releases/download/trt"`
`30`		`-$trt_name="TensorRT-7.1.3.4.Windows10.x86_64.cuda-11.0.cudnn8.0.zip"`
	`30`	`+$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"`
`31`	`31`
`32`	`32`	`if (-not (Test-Path "../${trt_name}"))`
`33`	`33`	`{`