Dockerfile.gpu-demo 318 字节
Newer Older
Y
yuyang18 已提交
1
FROM nvidia/cuda:7.5-cudnn5-devel-ubuntu14.04
Z
zhangjinchao01 已提交
2 3
MAINTAINER PaddlePaddle Dev Team <paddle-dev@baidu.com>
COPY build.sh /root/
Y
Yu Yang 已提交
4
ENV GIT_CHECKOUT=develop
Y
yuyang18 已提交
5 6
ENV WITH_GPU=ON
ENV IS_DEVEL=ON
Y
yuyang18 已提交
7 8 9
ENV WITH_DEMO=ON
ENV PIP_INSTALL_ARGS ""
ENV PIP_GENERAL_ARGS ""
Y
yuyang18 已提交
10
ENV USE_UBUNTU_MIRROR OFF
Y
Yu Yang 已提交
11
ENV WITH_AVX=ON
Z
zhangjinchao01 已提交
12
RUN cd /root/ && bash build.sh