parent
25aebf9879
commit
9aca46723e
5
Makefile
5
Makefile
|
@ -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"; \
|
||||
|
|
19
ubuntu-14.04/digits/Dockerfile
Normal file
19
ubuntu-14.04/digits/Dockerfile
Normal 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"]
|
Loading…
Reference in a new issue