Skip to content

Commit

Permalink
Removed torch7 dependencies and added keras (for now). Closes #1182 (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
brollb authored Sep 14, 2018
1 parent ec2701f commit e11abd4
Showing 1 changed file with 7 additions and 12 deletions.
19 changes: 7 additions & 12 deletions Dockerfile.worker
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
# This has torch and cuda support
FROM kaixhin/cuda-torch
FROM gw000/keras
MAINTAINER Brian Broll <brian.broll@gmail.com>

# install nodejs v6
RUN groupadd --gid 1000 node \
RUN apt-get update && apt-get install -y gnupg2 curl unzip && \
groupadd --gid 1000 node \
&& useradd --uid 1000 --gid node --shell /bin/bash --create-home node

# gpg keys listed at https://github.com/nodejs/node#release-team
Expand All @@ -18,7 +18,9 @@ RUN set -ex \
56730D5401028683275BD23C23EFEFE93C4CFFFE \
77984A986EBC2AA786BC0F66B01FBB92821C587A \
; do \
gpg --keyserver ha.pool.sks-keyservers.net --recv-keys "$key"; \
(gpg --keyserver ha.pool.sks-keyservers.net --recv-keys "$key" \
|| gpg --keyserver keyserver.pgp.com --recv-keys "$key" \
|| gpg --keyserver pgp.mit.edu --recv-keys "$key") \
done

ENV NPM_CONFIG_LOGLEVEL info
Expand All @@ -40,7 +42,7 @@ RUN mkdir /deepforge
ADD . /deepforge
WORKDIR /deepforge

RUN npm install
RUN npm install --production

RUN ln -s /deepforge/bin/deepforge /usr/local/bin

Expand All @@ -49,15 +51,8 @@ RUN deepforge config blob.dir /data/blob && \
deepforge config mongo.dir /data/db && \
deepforge config worker.cache.useBlob false && \
deepforge config worker.cache.dir /deepforge/worker-cache && \
deepforge config torch.dir /root/torch/ && \
git config --global user.email "deepforge-worker@deepforge.org" && \
git config --global user.name "deepforge-worker"

# Update torch
RUN apt-get update && apt-get install sudo wget && \
. /root/torch/install/bin/torch-activate && \
cd /root/torch/ && bash /root/torch/update.sh && \
deepforge update -t

ENTRYPOINT ["deepforge", "start", "--worker"]
CMD ["http://172.17.0.1:8888"]

0 comments on commit e11abd4

Please sign in to comment.