Update Dockerfile for GPU now that the branch has been merged (#3576)

* Remove cd and checkout * Updating comment

Update Dockerfile for GPU now that the branch has been merged (#3576)
* Remove cd and checkout * Updating comment
4c7c8fa7 · Karmel Allison · GitHub · 69c75593 · 4c7c8fa7 · 4c7c8fa7
Unverified Commit 4c7c8fa7 authored Mar 13, 2018 by Karmel Allison Committed by GitHub Mar 13, 2018
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

official/Dockerfile.cpu official/Dockerfile.cpu +1 -1

official/Dockerfile.gpu official/Dockerfile.gpu +3 -3

No files found.
--- a/official/Dockerfile.cpu
+++ b/official/Dockerfile.cpu
@@ -12,6 +12,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \

 RUN pip install tf-nightly

-# Checkout tensorflow/models and pin to a particular version
+# Checkout tensorflow/models at HEAD
 RUN git clone https://github.com/tensorflow/models.git /tensorflow_models

--- a/official/Dockerfile.gpu
+++ b/official/Dockerfile.gpu
@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \

 RUN pip install tf-nightly-gpu

-# Checkout tensorflow/models and pin to a particular version
-RUN git clone https://github.com/tensorflow/models.git /tensorflow_models && \
-    cd /tensorflow_models && git checkout feat/multi-gpu-resnet
+# Checkout tensorflow/models at HEAD
+RUN git clone https://github.com/tensorflow/models.git /tensorflow_models
+