Update Dockerfile

docker/Dockerfile

@@ -4,7 +4,8 @@ FROM tensorflow/tensorflow:1.15.0-gpu-py3
 # Set environment variables
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
 ENV DEBIAN_FRONTEND=noninteractive
-ENV PATH /opt/conda/bin:$PATH
+ENV PATH=/opt/conda/bin:$PATH
+ARG CUDA=12.2.2
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \