Add a DIGITS Dockerfile based on the caffe image

Closes: #19
This commit is contained in:
Felix Abecassis 2016-01-04 21:09:29 -08:00
parent 25aebf9879
commit 9aca46723e
2 changed files with 23 additions and 1 deletions

View file

@ -84,9 +84,12 @@ cudnn-runtime: $(CUDNN_RUNTIME_LATEST)
all-cudnn: $(CUDNN_VERSIONS) cudnn cudnn-devel cudnn-runtime
# caffe-nv images
caffe: $(OS)/caffe/Dockerfile
caffe: 7.0-cudnn4-runtime $(OS)/caffe/Dockerfile
$(DOCKER_BIN) build -t caffe $(OS)/caffe
digits: caffe $(OS)/digits/Dockerfile
$(DOCKER_BIN) build -t digits $(OS)/digits
push: all-cuda all-cudnn
if [ -z "$(USERNAME)" ]; then \
echo "Error: USERNAME not set"; \

View file

@ -0,0 +1,19 @@
FROM caffe
MAINTAINER NVIDIA CORPORATION <digits@nvidia.com>
ENV DIGITS_VERSION 3.0
LABEL com.nvidia.digits.version="3.0"
ENV DIGITS_PKG_VERSION 3.0.0~rc.3-1
RUN apt-get update && apt-get install -y --no-install-recommends --force-yes \
torch7-nv \
graphviz \
digits=$DIGITS_PKG_VERSION && \
rm -rf /var/lib/apt/lists/*
RUN mkdir -p /var/log/digits
RUN mkdir -p /usr/share/digits/digits/jobs
EXPOSE 34448
WORKDIR /usr/share/digits
ENTRYPOINT ["./digits-server"]