diff --git a/Dockerfile.worker b/Dockerfile.worker index a9bef503f..9aff680b6 100644 --- a/Dockerfile.worker +++ b/Dockerfile.worker @@ -1,9 +1,9 @@ -# This has torch and cuda support -FROM kaixhin/cuda-torch +FROM gw000/keras MAINTAINER Brian Broll # 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 @@ -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 @@ -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 @@ -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"]