Skip to content

Commit

Permalink
GPU: Docker Image Changes
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
airbrain-org committed Jan 30, 2020
1 parent 8a5c240 commit 04f9f6b
Show file tree
Hide file tree
Showing 4 changed files with 80 additions and 77 deletions.
3 changes: 2 additions & 1 deletion Dockerfile.cpu
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,8 @@ ARG userid=1000

RUN apt-get update && apt-get install -y \
graphviz \
libgraphviz-dev
libgraphviz-dev \
git

COPY ./requirements.txt /
RUN python3 -m pip install --upgrade pip
Expand Down
3 changes: 2 additions & 1 deletion Dockerfile.gpu
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,8 @@ ARG userid=1000

RUN apt-get update && apt-get install -y \
graphviz \
libgraphviz-dev
libgraphviz-dev \
git

COPY ./requirements.txt /
RUN python3 -m pip install --upgrade pip
Expand Down
3 changes: 2 additions & 1 deletion launch-docker-gpu.sh
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
# USAGE - ./launch-docker-gpu.sh {abs-path-to-GDL-code}
docker run --rm --runtime=nvidia --network=host -it -v $1:/GDL gdl-image
docker run --rm --runtime=nvidia -p 8888:8888 -it -v $1:/GDL gdl-image-gpu

148 changes: 74 additions & 74 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,75 +1,75 @@
absl-py==0.7.0
appnope==0.1.0
astor==0.7.1
backcall==0.1.0
bleach==3.1.0
cloudpickle==0.8.0
cycler==0.10.0
dask==1.1.1
decorator==4.3.2
defusedxml==0.5.0
entrypoints==0.3
gast==0.2.2
grpcio==1.18.0
h5py==2.9.0
ipykernel==5.1.0
ipython==7.3.0
ipython-genutils==0.2.0
ipywidgets==7.4.2
jedi==0.13.2
Jinja2==2.10
jsonschema==2.6.0
jupyter==1.0.0
jupyter-client==5.2.4
jupyter-console==6.0.0
jupyter-core==4.4.0
Keras==2.2.4
Keras-Applications==1.0.7
Keras-Preprocessing==1.0.9
absl-py
appnope
astor
backcall
bleach
cloudpickle
cycler
dask
decorator
defusedxml
entrypoints
gast
grpcio
h5py
ipykernel
ipython
ipython-genutils
ipywidgets
jedi
Jinja2
jsonschema
jupyter
jupyter-client
jupyter-console
jupyter-core
Keras
Keras-Applications
Keras-Preprocessing
git+https://www.github.com/keras-team/keras-contrib.git
kiwisolver==1.0.1
Markdown==3.0.1
MarkupSafe==1.1.0
matplotlib==3.0.2
mistune==0.8.4
music21==5.5.0
nbconvert==5.4.1
nbformat==4.4.0
networkx==2.2
notebook==5.7.4
numpy==1.16.1
pandas==0.24.2
pandocfilters==1.4.2
parso==0.3.4
pexpect==4.6.0
pickleshare==0.7.5
Pillow==5.4.1
prometheus-client==0.6.0
prompt-toolkit==2.0.9
protobuf==3.6.1
ptyprocess==0.6.0
pydot==1.4.1
Pygments==2.3.1
pyparsing==2.3.1
python-dateutil==2.8.0
pytz==2018.9
PyWavelets==1.0.1
PyYAML==3.13
pyzmq==18.0.0
qtconsole==4.4.3
scikit-image==0.14.2
scipy==1.2.1
Send2Trash==1.5.0
six==1.12.0
tensorboard==1.14.0
tensorflow==1.14.0
termcolor==1.1.0
terminado==0.8.1
testpath==0.4.2
toolz==0.9.0
tornado==5.1.1
traitlets==4.3.2
wcwidth==0.1.7
webencodings==0.5.1
Werkzeug==0.14.1
widgetsnbextension==3.4.2
kiwisolver
Markdown
MarkupSafe
matplotlib
mistune
music21
nbconvert
nbformat
networkx
notebook
numpy
pandas
pandocfilters
parso
pexpect
pickleshare
Pillow
prometheus-client
prompt-toolkit
protobuf
ptyprocess
pydot
Pygments
pyparsing
python-dateutil
pytz
PyWavelets
PyYAML
pyzmq
qtconsole
scikit-image
scipy
Send2Trash
six
tensorboard
tensorflow
termcolor
terminado
testpath
toolz
tornado
traitlets
wcwidth
webencodings
Werkzeug
widgetsnbextension

0 comments on commit 04f9f6b

Please sign in to comment.