Skip to content

Commit 3128040

Browse files
Merge pull request #4 from yifeitao/patch-1
fix cuda version & tensorflow version
2 parents 0a349ee + ab364ee commit 3128040

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docker/gpu/Dockerfile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM nvidia/cuda:10.1-cudnn7-devel
1+
FROM nvidia/cuda:10.0-cudnn7-devel
22

33
RUN apt-get update && apt-get upgrade -y
44

@@ -18,7 +18,7 @@ RUN DEBIAN_FRONTEND=noninteractive apt-get install -y protobuf-compiler python3-
1818
pip3 install jupyter && \
1919
pip3 install matplotlib && \
2020
pip3 install Pillow && \
21-
pip3 install tensorflow==1.14.0
21+
pip3 install tensorflow-gpu==1.14.0
2222

2323
# Install pycocoapi
2424
RUN git clone --depth 1 https://github.com/cocodataset/cocoapi.git && \
@@ -47,4 +47,4 @@ ENV PYTHONPATH=$PYTHONPATH:/models/research
4747

4848
# Entrypoint
4949
WORKDIR /models/research/object_detection
50-
ENTRYPOINT ["jupyter", "notebook", "--ip=0.0.0.0", "--no-browser", "--allow-root"]
50+
ENTRYPOINT ["jupyter", "notebook", "--ip=0.0.0.0", "--no-browser", "--allow-root"]

0 commit comments

Comments
 (0)