fix: update cuda env vars

Author: PGerardi

Dockerfile

@@ -4,6 +4,7 @@ ARG CUDA_VERSION=11.8
 ARG CUDNN_VERSION=8.9.4.25
 ARG TENSORRT_VERSION=8.6.1.6
 FROM python:$PYTHON_VERSION-slim AS base
+ENV CUDA_VERSION=11.8
 ENV CUDNN_VERSION=8.9.4.25
 ENV TENSORRT_VERSION=8.6.1.6
 # Install wget.