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

AmusementClub · Oct 1, 2024 · 396996f · 396996f
1 parent 4fa2a5e
commit 396996f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/linux-trt.yml b/.github/workflows/linux-trt.yml
@@ -35,7 +35,8 @@ jobs:
         wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
         sudo dpkg -i cuda-keyring_1.1-1_all.deb
         sudo apt-get update
-        sudo apt-get install -y --no-install-recommends cuda-nvcc-12-6 cuda-cudart-dev-12-6 libnvinfer-dev=10.4.0.26-1+cuda12.6 libnvinfer-headers-dev=10.4.0.26-1+cuda12.6
+        export TRT_VER=10.4.0.26-1+cuda12.6
+        sudo apt-get install -y --no-install-recommends cuda-nvcc-12-6 cuda-cudart-dev-12-6 libnvinfer-dev=${TRT_VER} libnvinfer-headers-dev=${TRT_VER} libnvinfer10=${TRT_VER}
         echo "PATH=/usr/local/cuda/bin${PATH:+:${PATH}}" >> $GITHUB_ENV
         echo "CUDA_PATH=/usr/local/cuda" >> $GITHUB_ENV
         echo "LD_LIBRARY_PATH=/usr/local/cuda/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}" >> $GITHUB_ENV