diff --git a/docker/tools/visualize/Dockerfile.tf1_3_gpu b/docker/tools/visualize/Dockerfile.tf1_3_gpu
index 546b8ec..0778c8f 100644
--- a/docker/tools/visualize/Dockerfile.tf1_3_gpu
+++ b/docker/tools/visualize/Dockerfile.tf1_3_gpu
@@ -172,4 +172,5 @@ VOLUME /root/corpora_out
 #ENTRYPOINT ["/root/recursive-embedding/set-user-with-folder.sh", "/root/train", "python", "train_fold.py"]
 
 # pass as env variable
+ARG NVIDIA_VISIBLE_DEVICES=0
 ENV NVIDIA_VISIBLE_DEVICES="$NVIDIA_VISIBLE_DEVICES"