fixed docker-compose usage for training

ArneBinder · Apr 26, 2018 · 56cc45a · 56cc45a
1 parent bba2897
commit 56cc45a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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"