Skip to content

Commit

Permalink
fixed docker-compose usage for training
Browse files Browse the repository at this point in the history
  • Loading branch information
@ArneBinder
ArneBinder committed Apr 26, 2018
1 parent 56cc45a commit 623a1d8
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions docker/tools/visualize/Dockerfile.tf1_3_gpu
View file
Original file line number Diff line number Diff line change
Expand Up @@ -172,5 +172,6 @@ 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"
ARG _NVIDIA_VISIBLE_DEVICES=0
RUN echo NVIDIA_VISIBLE_DEVICES="$_NVIDIA_VISIBLE_DEVICES"
ENV NVIDIA_VISIBLE_DEVICES="$_NVIDIA_VISIBLE_DEVICES"

0 comments on commit 623a1d8

Please sign in to comment.