Dockerfile.ngraph.centos74_gpu 1.29 KB
Newer Older
1
# Environment to build and unit-test ngraph on centos74 for GPU backend
2 3
# with gcc 4.8.5
# with python 2.7
4
# with pre-built cmake3
5

6
FROM nvidia/cuda:8.0-cudnn7-devel-centos7
7 8 9 10 11 12 13 14 15

RUN yum -y update && \
    yum -y --enablerepo=extras install epel-release && \
    yum -y install \
    gcc gcc-c++ \
    cmake3 make \
    git \
    wget patch diffutils zlib-devel ncurses-devel libtinfo-dev \
    python python-devel python-setuptools \
16
    doxygen graphviz \
17 18 19 20
    which \
    'perl(Data::Dumper)'

RUN ln -s /usr/bin/cmake3 /usr/bin/cmake
21
RUN ln -s /usr/local/cuda/include/cudnn.h /usr/local/cuda/include/cudnn_v7.h
22 23 24 25 26 27 28

RUN cmake --version
RUN make --version
RUN gcc --version
RUN c++ --version

RUN easy_install pip
29
RUN pip install virtualenv==16.7.10
30 31 32 33 34 35 36 37 38 39 40 41 42

# Install some pip packages

# need to use sphinx version 1.6 to build docs
# installing with apt-get install python-sphinx installs sphinx version 1.3.6 only
# added install for python-pip above and
# installed sphinx with pip to get the updated version 1.6.5
# allows for make html build under the doc/source directory as an interim build process
RUN pip install sphinx

# breathe package required to build documentation
RUN pip install breathe

43 44 45
# need numpy to successfully build docs for python_api
RUN pip install numpy

46 47
# RUN python3 -m pip install m2r

48
WORKDIR /home