Commit 3128040

authored

Merge pull request #4 from yifeitao/patch-1

fix cuda version & tensorflow version

2 parents 0a349ee + ab364ee commit 3128040Copy full SHA for 3128040

File tree

-3

lines changed

-3

lines changed

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM nvidia/cuda:10.1-cudnn7-devel`
	`1`	`+FROM nvidia/cuda:10.0-cudnn7-devel`
`2`	`2`
`3`	`3`	`RUN apt-get update && apt-get upgrade -y`
`4`	`4`
`@@ -18,7 +18,7 @@ RUN DEBIAN_FRONTEND=noninteractive apt-get install -y protobuf-compiler python3-`
`18`	`18`	`pip3 install jupyter && \`
`19`	`19`	`pip3 install matplotlib && \`
`20`	`20`	`pip3 install Pillow && \`
`21`		`- pip3 install tensorflow==1.14.0`
	`21`	`+ pip3 install tensorflow-gpu==1.14.0`
`22`	`22`
`23`	`23`	`# Install pycocoapi`
`24`	`24`	`RUN git clone --depth 1 https://github.com/cocodataset/cocoapi.git && \`
`@@ -47,4 +47,4 @@ ENV PYTHONPATH=$PYTHONPATH:/models/research`
`47`	`47`
`48`	`48`	`# Entrypoint`
`49`	`49`	`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"]`

Comments

(0)