Dockerfile.gpu-devel 294 字节
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
yuyang18 已提交
4 5
ENV WITH_GPU=ON
ENV IS_DEVEL=ON
Y
yuyang18 已提交
6 7 8
ENV WITH_DEMO=OFF
ENV PIP_INSTALL_ARGS ""
ENV PIP_GENERAL_ARGS ""
Y
yuyang18 已提交
9
ENV USE_UBUNTU_MIRROR OFF
Y
Yu Yang 已提交
10
ENV WITH_AVX=ON
Z
zhangjinchao01 已提交
11
RUN cd /root/ && bash build.sh