Skip to content
Permalink
Browse files

ADD: dockerfiles + scripts

  • Loading branch information...
madisonmay committed Jul 6, 2018
1 parent 8cdc2a2 commit 16770a4dc78c464fedae2ad7e8e765cf2b174e05
Showing with 37 additions and 2 deletions.
  1. +1 −0 .dockerignore
  2. +23 −0 docker/Dockerfile
  3. +1 −0 docker/build_docker.sh
  4. +1 −0 docker/start_docker.sh
  5. +1 −1 finetune/lm_base.py
  6. +10 −0 requirements.txt
  7. +0 −1 setup.py
@@ -0,0 +1 @@
tests/saved-models/*
@@ -0,0 +1,23 @@
FROM tensorflow/tensorflow:1.6.0-gpu-py3
LABEL Author="Ben Townsend, Madison May"

# tensorboard
EXPOSE 6006

# nvidia-docker 1.0
LABEL com.nvidia.volumes.needed="nvidia_driver"
LABEL com.nvidia.cuda.version="${CUDA_VERSION}"

# nvidia-container-runtime
ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
ENV NVIDIA_REQUIRE_CUDA "cuda>=8.0"

RUN apt-get update
RUN mkdir /Finetune
ADD requirements.txt /Finetune/requirements.txt
RUN pip3 install -r /Finetune/requirements.txt
RUN python3 -m spacy download en
ENV LANG C.UTF-8
WORKDIR /Finetune
CMD ["sleep","infinity"]
@@ -0,0 +1 @@
docker build -t finetune --file ./docker/Dockerfile .
@@ -0,0 +1 @@
docker run --runtime=nvidia -d -v $(pwd):/Finetune finetune
@@ -352,7 +352,7 @@ def _build_model(self, n_updates_total, target_dim, train=True):
"""
if not self.is_trained or train != self.train:
# reconstruct graph to include/remove dropout
# #if `train` setting has changed
# if `train` setting has changed
self._construct_graph(n_updates_total, target_dim, train=train)

# Optionally load saved model
@@ -0,0 +1,10 @@
pandas>=0.23.1
IndicoIo>=1.1.5
enso>=0.1.6
tqdm>=4.0.0
numpy>=1.13.0
scikit-learn>=0.18.0
ftfy>=4.4.0
spacy>=2.0.0
msgpack-numpy>=0.4.1
nose>=1.3.7
@@ -13,7 +13,6 @@
"tqdm>=4.0.0",
"numpy>=1.13.0",
"scikit-learn>=0.18.0",
"joblib>=0.11",
"ftfy>=4.4.0",
"spacy>=2.0.0",
"msgpack-numpy>=0.4.1",

0 comments on commit 16770a4

Please sign in to comment.
You can’t perform that action at this time.