Skip to content

GitLab

Unverified Commit 4c7c8fa7 authored by Karmel Allison's avatar Karmel Allison Committed by GitHub
Browse files

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

* Remove cd and checkout

* Updating comment
parent 69c75593
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
official/Dockerfile.cpu
View file @ 4c7c8fa7
...@@ -12,6 +12,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ ...@@ -12,6 +12,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
RUN pip install tf-nightly 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 RUN git clone https://github.com/tensorflow/models.git /tensorflow_models
official/Dockerfile.gpu
View file @ 4c7c8fa7
...@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ ...@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
RUN pip install tf-nightly-gpu RUN pip install tf-nightly-gpu
# 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 && \ RUN git clone https://github.com/tensorflow/models.git /tensorflow_models
cd /tensorflow_models && git checkout feat/multi-gpu-resnet
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment