Dockerfile.gpu 1.97 KB
Newer Older
1
FROM nvidia/cuda:8.0-cudnn7-devel
2 3

# Base scripts
4
RUN apt-get update --fix-missing
5

6
COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh
7
RUN bash /install/ubuntu_install_core.sh
8 9

COPY install/ubuntu_install_python.sh /install/ubuntu_install_python.sh
10
RUN bash /install/ubuntu_install_python.sh
11 12

COPY install/ubuntu_install_llvm.sh /install/ubuntu_install_llvm.sh
13
RUN bash /install/ubuntu_install_llvm.sh
14 15

COPY install/ubuntu_install_opencl.sh /install/ubuntu_install_opencl.sh
16
RUN bash /install/ubuntu_install_opencl.sh
17 18

COPY install/ubuntu_install_iverilog.sh /install/ubuntu_install_iverilog.sh
19 20
RUN bash /install/ubuntu_install_iverilog.sh

21 22 23
COPY install/ubuntu_install_python_package.sh /install/ubuntu_install_python_package.sh
RUN bash /install/ubuntu_install_python_package.sh

24 25 26
COPY install/ubuntu_install_sphinx.sh /install/ubuntu_install_sphinx.sh
RUN bash /install/ubuntu_install_sphinx.sh

27 28 29 30
# Fix recommonmark to latest version
RUN git clone https://github.com/rtfd/recommonmark
RUN cd recommonmark; python setup.py install

Tianqi Chen committed
31 32 33
COPY install/ubuntu_install_java.sh /install/ubuntu_install_java.sh
RUN bash /install/ubuntu_install_java.sh

34 35 36
COPY install/ubuntu_install_nodejs.sh /install/ubuntu_install_nodejs.sh
RUN bash /install/ubuntu_install_nodejs.sh

37 38 39
COPY install/ubuntu_install_rocm.sh /install/ubuntu_install_rocm.sh
RUN bash /install/ubuntu_install_rocm.sh

40 41 42
# Enable doxygen for c++ doc build
RUN apt-get install -y doxygen graphviz

43
# Environment variables
44
ENV PATH=/node_modules/.bin:${PATH}
45 46 47 48 49 50
ENV PATH=/usr/local/nvidia/bin:${PATH}
ENV PATH=/usr/clang+llvm-4.0.0-x86_64-linux-gnu-ubuntu-14.04/bin:${PATH}
ENV PATH=/usr/local/cuda/bin:${PATH}
ENV CPLUS_INCLUDE_PATH=/usr/local/cuda/include:${CPLUS_INCLUDE_PATH}
ENV C_INCLUDE_PATH=/usr/local/cuda/include:${C_INCLUDE_PATH}
ENV LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/nvidia/lib64:${LIBRARY_PATH}
Tianqi Chen committed
51
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}
52
ENV LD_LIBRARY_PATH=/opt/rocm/lib:${LD_LIBRARY_PATH}