chore: use parameters instead of hard coded values

Author: PGerardi

Dockerfile

@@ -1,15 +1,16 @@
 # syntax=docker/dockerfile:1
 ARG DEBIAN_FRONTEND=noninteractive
 ARG PYTHON_VERSION=3.8
+
+FROM python:$PYTHON_VERSION-slim AS base
+
 ARG NVARCH=x86_64
 ARG CUDA=11.8
 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
 
-FROM python:$PYTHON_VERSION-slim AS base
-
 # 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.