Skip to content

Commit

Permalink
Adding utility scripts.
Browse files Browse the repository at this point in the history
  • Loading branch information
airbrain-org committed Feb 16, 2020
1 parent 04f9f6b commit 8ea0e65
Show file tree
Hide file tree
Showing 3 changed files with 33 additions and 0 deletions.
30 changes: 30 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
# See all tag variants at https://hub.docker.com/r/tensorflow/tensorflow/tags/
# build with `ln -sf Dockerfile.gpu Dockerfile && docker build --network=host -t {container-name} .`
# NOTE(jwd) - if you wish to use this implementation, you must install nvidia-docker v2.0
# see https://github.com/nvidia/nvidia-docker/wiki/Installation-(version-2.0) for steps
FROM tensorflow/tensorflow:latest-gpu-py3-jupyter

## modify below
ARG username=gdl
ARG groupid=1000
ARG userid=1000
## end modify

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

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

# -m option creates a fake writable home folder for Jupyter.
RUN groupadd -g $groupid $username \
&& useradd -m -r -u $userid -g $username $username
USER $username

VOLUME ["/GDL"]
WORKDIR /GDL

CMD ["jupyter", "notebook", "--no-browser", "--ip=0.0.0.0", "/GDL"]
1 change: 1 addition & 0 deletions build.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
docker build --network=host -t gdl-image-gpu .
2 changes: 2 additions & 0 deletions run.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
./launch-docker-gpu.sh ~/develop/GDL_code

0 comments on commit 8ea0e65

Please sign in to comment.