.github/workflows/linux-trt.yml: fix trt installation

AmusementClub · Mar 28, 2024 · 8086bed · 8086bed
1 parent 0c4d8ca
commit 8086bed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/linux-trt.yml b/.github/workflows/linux-trt.yml
@@ -42,7 +42,7 @@ jobs:
 
     - name: Setup tensorrt
       run: |
-        sudo apt-get install -y libnvinfer-dev=8.6.1.6-1+cuda12.0
+        sudo apt-get install -y libnvinfer-dev=8.6.1.6-1+cuda12.0 libnvinfer-headers-dev=8.6.1.6-1+cuda12.0
 
     - name: Configure
       run: cmake -S . -B build -G Ninja -LA