chore: use parameters instead of hard coded values

Author: PGerardi

Dockerfile

@@ -10,6 +10,10 @@ ARG CUDNN_VERSION=8.6.0.163
 ARG NV_CUDA_CUDART_VERSION=11.8.89-1
 ARG NV_CUDA_COMPAT_PACKAGE=cuda-compat-11-8
 ARG TF_TENSORRT_VERSION=8.4.3
+# CUDDN
+ARG CUDNN_VERSION=8.6.0.163-1+cuda11.8
+# TensorRT
+ARG LIBINVER_VERSION=8.4.3-1+cuda11.6
 
 # NVIDIA: https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.8.0/ubuntu2204/base/Dockerfile
 # specify the version of the CUDA Toolkit to use and the which driver versions are compatible for each brand of GPU.
@@ -19,7 +23,7 @@ ENV NVARCH=$NVARCH \
     CUDNN_VERSION=$CUDNN_VERSION \
     NV_CUDA_CUDART_VERSION=$NV_CUDA_CUDART_VERSION \
     NV_CUDA_COMPAT_PACKAGE=$NV_CUDA_COMPAT_PACKAGE \
-    TF_TENSORRT_VERSION=$TF_TENSORRT_VERSION \
+    LIBINVER_VERSION=$LIBINVER_VERSION \
     NVIDIA_REQUIRE_CUDA="cuda>=$CUDA brand=tesla,driver>=450,driver<451 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=510,driver<511 brand=unknown,driver>=510,driver<511 brand=nvidia,driver>=510,driver<511 brand=nvidiartx,driver>=510,driver<511 brand=geforce,driver>=510,driver<511 brand=geforcertx,driver>=510,driver<511 brand=quadro,driver>=510,driver<511 brand=quadrortx,driver>=510,driver<511 brand=titan,driver>=510,driver<511 brand=titanrtx,driver>=510,driver<511 brand=tesla,driver>=515,driver<516 brand=unknown,driver>=515,driver<516 brand=nvidia,driver>=515,driver<516 brand=nvidiartx,driver>=515,driver<516 brand=geforce,driver>=515,driver<516 brand=geforcertx,driver>=515,driver<516 brand=quadro,driver>=515,driver<516 brand=quadrortx,driver>=515,driver<516 brand=titan,driver>=515,driver<516 brand=titanrtx,driver>=515,driver<516"
 
 #  Updates the package index and installs the necessarys packages to add the CUDA repository, including `gnupg2`, `curl`, and `ca-certificates`. 
@@ -47,9 +51,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
         libcusolver-${CUDA%.*}-${CUDA#*.} \
         libcusparse-${CUDA%.*}-${CUDA#*.} \
         libtool \
-        libcudnn8=${CUDNN_VERSION}-1+cuda${NV_CUDA_CUDART_VERSION} \
-        libnvinfer8=${TF_TENSORRT_VERSION}-1+cuda${CUDA} \
-        libnvinfer-plugin8=${TF_TENSORRT_VERSION}-1+cuda${CUDA} \
+        libcudnn8=${CUDNN_VERSION}\
+        libnvinfer8=${LIBINVER_VERSION} \
+        libnvinfer-plugin8=${LIBINVER_VERSION} \
         build-essential \
         pkg-config \
         software-properties-common \