Skip to content

Commit 04f9f6b

Browse files
committed
GPU: Docker Image Changes
Modified requirements.txt in the Dockerfile.gpu to remove the version information after each python package. This was necessary to allow tensorflow/tensorflow:latest-gpu-py3-jupyter to successfully access the GPU's on the host system.
1 parent 8a5c240 commit 04f9f6b

File tree

4 files changed

+80
-77
lines changed

4 files changed

+80
-77
lines changed

Dockerfile.cpu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,8 @@ ARG userid=1000
1212

1313
RUN apt-get update && apt-get install -y \
1414
graphviz \
15-
libgraphviz-dev
15+
libgraphviz-dev \
16+
git
1617

1718
COPY ./requirements.txt /
1819
RUN python3 -m pip install --upgrade pip

Dockerfile.gpu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,8 @@ ARG userid=1000
1212

1313
RUN apt-get update && apt-get install -y \
1414
graphviz \
15-
libgraphviz-dev
15+
libgraphviz-dev \
16+
git
1617

1718
COPY ./requirements.txt /
1819
RUN python3 -m pip install --upgrade pip

launch-docker-gpu.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
# USAGE - ./launch-docker-gpu.sh {abs-path-to-GDL-code}
2-
docker run --rm --runtime=nvidia --network=host -it -v $1:/GDL gdl-image
2+
docker run --rm --runtime=nvidia -p 8888:8888 -it -v $1:/GDL gdl-image-gpu
3+

requirements.txt

Lines changed: 74 additions & 74 deletions
Original file line numberDiff line numberDiff line change
@@ -1,75 +1,75 @@
1-
absl-py==0.7.0
2-
appnope==0.1.0
3-
astor==0.7.1
4-
backcall==0.1.0
5-
bleach==3.1.0
6-
cloudpickle==0.8.0
7-
cycler==0.10.0
8-
dask==1.1.1
9-
decorator==4.3.2
10-
defusedxml==0.5.0
11-
entrypoints==0.3
12-
gast==0.2.2
13-
grpcio==1.18.0
14-
h5py==2.9.0
15-
ipykernel==5.1.0
16-
ipython==7.3.0
17-
ipython-genutils==0.2.0
18-
ipywidgets==7.4.2
19-
jedi==0.13.2
20-
Jinja2==2.10
21-
jsonschema==2.6.0
22-
jupyter==1.0.0
23-
jupyter-client==5.2.4
24-
jupyter-console==6.0.0
25-
jupyter-core==4.4.0
26-
Keras==2.2.4
27-
Keras-Applications==1.0.7
28-
Keras-Preprocessing==1.0.9
1+
absl-py
2+
appnope
3+
astor
4+
backcall
5+
bleach
6+
cloudpickle
7+
cycler
8+
dask
9+
decorator
10+
defusedxml
11+
entrypoints
12+
gast
13+
grpcio
14+
h5py
15+
ipykernel
16+
ipython
17+
ipython-genutils
18+
ipywidgets
19+
jedi
20+
Jinja2
21+
jsonschema
22+
jupyter
23+
jupyter-client
24+
jupyter-console
25+
jupyter-core
26+
Keras
27+
Keras-Applications
28+
Keras-Preprocessing
2929
git+https://www.github.com/keras-team/keras-contrib.git
30-
kiwisolver==1.0.1
31-
Markdown==3.0.1
32-
MarkupSafe==1.1.0
33-
matplotlib==3.0.2
34-
mistune==0.8.4
35-
music21==5.5.0
36-
nbconvert==5.4.1
37-
nbformat==4.4.0
38-
networkx==2.2
39-
notebook==5.7.4
40-
numpy==1.16.1
41-
pandas==0.24.2
42-
pandocfilters==1.4.2
43-
parso==0.3.4
44-
pexpect==4.6.0
45-
pickleshare==0.7.5
46-
Pillow==5.4.1
47-
prometheus-client==0.6.0
48-
prompt-toolkit==2.0.9
49-
protobuf==3.6.1
50-
ptyprocess==0.6.0
51-
pydot==1.4.1
52-
Pygments==2.3.1
53-
pyparsing==2.3.1
54-
python-dateutil==2.8.0
55-
pytz==2018.9
56-
PyWavelets==1.0.1
57-
PyYAML==3.13
58-
pyzmq==18.0.0
59-
qtconsole==4.4.3
60-
scikit-image==0.14.2
61-
scipy==1.2.1
62-
Send2Trash==1.5.0
63-
six==1.12.0
64-
tensorboard==1.14.0
65-
tensorflow==1.14.0
66-
termcolor==1.1.0
67-
terminado==0.8.1
68-
testpath==0.4.2
69-
toolz==0.9.0
70-
tornado==5.1.1
71-
traitlets==4.3.2
72-
wcwidth==0.1.7
73-
webencodings==0.5.1
74-
Werkzeug==0.14.1
75-
widgetsnbextension==3.4.2
30+
kiwisolver
31+
Markdown
32+
MarkupSafe
33+
matplotlib
34+
mistune
35+
music21
36+
nbconvert
37+
nbformat
38+
networkx
39+
notebook
40+
numpy
41+
pandas
42+
pandocfilters
43+
parso
44+
pexpect
45+
pickleshare
46+
Pillow
47+
prometheus-client
48+
prompt-toolkit
49+
protobuf
50+
ptyprocess
51+
pydot
52+
Pygments
53+
pyparsing
54+
python-dateutil
55+
pytz
56+
PyWavelets
57+
PyYAML
58+
pyzmq
59+
qtconsole
60+
scikit-image
61+
scipy
62+
Send2Trash
63+
six
64+
tensorboard
65+
tensorflow
66+
termcolor
67+
terminado
68+
testpath
69+
toolz
70+
tornado
71+
traitlets
72+
wcwidth
73+
webencodings
74+
Werkzeug
75+
widgetsnbextension

0 commit comments

Comments
 (0)