diff --git a/docker/Dockerfile b/docker/Dockerfile
index 3f852bad004b0d7048a9b30974000bda5b221448..978613a0d46cfe8720fb75accd13d2d054dfbc75 100644
--- a/docker/Dockerfile
+++ b/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 \